当运行下面代码时出现下面的错误ifiteration%10==0:print("===>Epoch[{}]({}/{}):Loss:{:.10f}".format(epoch,iteration,len(training_data_loader),loss.data[0]))修改建议:去掉loss.data[0]后面的[0]。ifiteration%10==0:print("===>Epoch[{}]({}/{}):Loss:{:.10f}".format(epoch,iteration,len(training_data_loader),loss.data))原因:loss是0维张量创建一个
对于3维矩阵,dim为-1时 与dim为2时的效果是一样的。dim为0时 从0维度,下图 是三维实例 图的目的是 可以由一个想象的空间。下面代码与上图关系不大>>>ab=torch.tensor([[[0,1,2,3],[1,2,3,4]],[[2,3,4,5],[4,5,6,7]],[[5,6,7,8],[6,7,8,9]]])>>>abtensor([[[0,1,2,3],[1,2,3,4]],[[2,3,4,5],[4,5,6,7]],[[5,6,7,8],[6,7,8,9]]])>>>ab.shape#ab的形状尺寸torch.Size([3,2,4])>>>ab.size()#ab的