python运行报错RuntimeError: view size is not compatible with input tensor‘s size and stride (at least..)

岳麓山下敲代码

于 2022-10-11 08:52:15 发布

阅读量655

点赞数

CC 4.0 BY-SA版权

分类专栏：研一啦文章标签： numpy

1 篇文章

订阅专栏

本文介绍了一种在PyTorch中遇到的Tensor不连续问题及其解决方案。问题在于使用view()时，若Tensor元素地址不连续，则会引发错误。文中提供了一个简单的修复方法：在view()前添加contiguous()确保Tensor元素地址连续。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个问题遇到了，一定要记录一下：

问题出现原因：

出现这个原因主要就是因为view()需要Tensor中的元素地址是连续的，因为可能出现Tensor不连续的情况，所以需要写代码的时候人为保持连续。

这个原来的代码是：

return self.weights[device].index_select(0, positions.view(-1)).view(bsz, seq_len, -1).detach()

在view前加入： contiguous()

return self.weights[device].index_select(0, positions.contiguous().view(-1)).contiguous().view(bsz, seq_len, -1).detach()