AttributeError: 'torch.FloatTensor' object has no attribute 'shape'
2018-08-16 12-54-35屏幕截图.png原因是torch的张量tensor,没有.shape属性。解决办法是将这个张量改成numpy数组,weight1 = weight.numpy()
2018-08-16 12-55-58屏幕截图.png补:发现用conda下载的torch版本比较旧,导致了这些乱七八糟的问题,后来者看到,更新一下torch版本就行。怎么更新看这里https://pytorch.org/previous-versions/
我用pip更新的
网友评论