pytorch tensor.contiguous()
时间: 2024-04-07 08:30:55 浏览: 135
pytorch实现Tensor变量之间的转换
5星 · 资源好评率100%
在 PyTorch 中,当一个张量不是按照内存中连续的方式存储时,它被称为非连续性张量。在这种情况下,我们可以使用 `tensor.contiguous()` 方法来返回一个连续的张量。
具体来说,当一个张量的内存不是按照连续方式存储时,这个张量的元素在内存中的存储位置可能是不连续的。这种情况下,我们无法直接对该张量进行一些操作,比如进行矩阵乘法等操作。因此,我们需要调用 `contiguous()` 方法来将该张量复制到一个连续的内存块中,使得元素在内存中的存储位置变得连续。这样,我们就可以对它进行操作了。
需要注意的是,`contiguous()` 方法会返回一个新的张量,而不是修改原有的张量。因此,如果需要修改原有的张量,需要将返回值再次赋值给原变量。
阅读全文