报错:AttributeError: 'torch.FloatTensor' object has no attribute 'shape'

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更新的

你可能感兴趣的:(报错:AttributeError: 'torch.FloatTensor' object has no attribute 'shape')