pytorch TRANSPOSE改变内存布局还是只改变视图?

yshpjwxd  于 2023-06-06  发布在  其他
关注(0)|答案(1)|浏览(429)

https://pytorch.org/docs/stable/tensors.html#torch.Tensor.T 说Tensor.T
返回此Tensor的视图,并反转其维度。
我想知道TORCH.TRANSPOSE是否相同。
更新
发现这篇文章有用What does .contiguous() do in PyTorch?

sqxo8psd

sqxo8psd1#

实际上在doc中提到:
如果input是一个stridedTensor,那么结果输出Tensor与inputTensor共享其底层存储,因此更改一个的内容将更改另一个的内容。
如果输入是稀疏Tensor,则结果输出Tensor不与输入Tensor共享底层存储。
因此,如果输入是一个跨步Tensor,输出将是一个视图;对于稀疏Tensor,它是复制的。

相关问题