RuntimeError:视图大小与输入张量的大小和步幅不兼容(至少一维跨越两个连续的子空间)

Bil*_*mas 13 python computer-vision deep-learning pytorch

我正在使用 Pytorch。我在评估模型时遇到了这个运行时错误。知道如何解决这个问题吗?

Bil*_*mas 17

解决方案:只需将view()函数替换为reshape()按照错误中的建议

我想这与张量在内存中的存储方式有关。


sta*_*iet 8

解决方案2:将 更改.view(...).contiguous().view(...)

实际上,如果需要的话,“reshape()”在幕后的作用与 contigious() 类似。