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

原因是torch的张量tensor,没有.shape属性。解决办法是将这个张量改成numpy数组,weight1 = weight.numpy()

补:发现用conda下载的torch版本比较旧,导致了这些乱七八糟的问题,后来者看到,更新一下torch版本就行。怎么更新看这里https://pytorch.org/previous-versions/
我用pip更新的
网友评论