python运行报错RuntimeError: view size is not compatible with input tensor’s size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(…) instead.

啊啊啊啊,出现这个错误已经不止一次了,但是每次改完过一段时间又忘记了,我觉得有必要给它记个笔记,以供自己后面复习。

报错代码如图:
在这里插入图片描述

根据报错的上方找到代码具体出错位置
出现这个原因主要就是因为view()需要Tensor中的元素地址是连续的,因为可能出现Tensor不连续的情况,所以修改方法为:
在.view前加.contiguous(),使其变为连续就ok。
eg.我的如图所示:
在这里插入图片描述

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐