2021年8月12日
expected Long (got Float)
摘要： 不同类型的Tensor之间不能互转 torch.LongTensor(labels) 解决方法：先转化成numpy labels=torch.LongTensor(labels.numpy()) 阅读全文
posted @ 2021-08-12 20:32 吴莫愁258 阅读(438) 评论(0) 推荐(0) 编辑
降维和升维
摘要： unsqueeze（arg）是增添第arg个维度为1，以插入的形式填充 squeeze（arg）是删除第arg个维度(如果当前维度不为1，则不会进行删除) 阅读全文
posted @ 2021-08-12 20:27 吴莫愁258 阅读(93) 评论(0) 推荐(0) 编辑