0
点赞
收藏
分享

微信扫一扫

PyTorch: tensor操作(一) contiguous

_LEON_ 2022-02-13 阅读 48

本文目录

tensor在内存中的存储

tensor在内存中的存储包含信息区和存储区

  • **信息区(Tensor)**包含tensor的形状size,步长stride,数据类型type等
  • **存储区(Storage)**包含存储的数据

contiguous

contiguous

此函数的用法很简单tensor.contiguous()即可

那么什么是连续内存的tensor呢?这就需要先明白tensor在内存中的存储方式

  • 对于刚创建的tensor,他们的数据在内存中是连续的
tensor = torch.rand((2, 3))
print(tensor.is_contiguous())  # True
  • 当我们对tensor进行view,transpose,permute等操作时,他们没有开辟新内存,也没有改变
举报

相关推荐

0 条评论