当前位置:   article > 正文

pytorch中contiguous()_pytorch contiguois

pytorch contiguois

pytorch中contiguous()

调用view之前最好先contiguous
x.contiguous().view() 
 
因为view需要tensor的内存是整块的 

contiguous:view只能用在contiguous的variable上。如果在view之前用了transpose, permute等,需要用contiguous()来返回一个contiguous copy。 
一种可能的解释是: 
有些tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式。 
判断是否contiguous用torch.Tensor.is_contiguous()函数。

  1. import torch
  2. x = torch.ones(10, 10)
  3. x.is_contiguous() # True
  4. x.transpose(0, 1).is_contiguous() # False
  5. x.transpose(0, 1).contiguous().is_contiguous() # True

pytorch的最新版本0.4版本中,增加了torch.reshape(), 这与 numpy.reshape 的功能类似。它大致相当于 tensor.contiguous().view()

https://blog.csdn.net/u011394059/article/details/78664694 
https://blog.csdn.net/guotong1988/article/details/78730078 
https://blog.csdn.net/u011276025/article/details/73826562 
中的Bug3

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/511981
推荐阅读
相关标签
  

闽ICP备14008679号