标签：__ tensor contiguous torch PyTorch print myArr

contiguous()——把tensor变成在内存中连续分布的形式

需要变成连续分布的情况：

contiguous：view只能用在contiguous的variable上。如果在view之前用了transpose, permute等，需要用contiguous()来返回一个contiguous copy。

import torch
import numpy as nn

def main():
    
    myArr = np.array([[[1,2,3],
                      [4,5,6]]])
    print(myArr.shape)
    print(myArr.size)
    myTsr = torch.tensor(myArr)
    print(myTsr)
    print(myTsr.shape)
    myTsr2 = myTsr.permute(2,0,1)
    print(myTsr2)
    print(myTsr2.shape)
    myTsr3 = myTsr2.contiguous().view(-1,2)
    print(myTsr3)
    print(myTsr3.shape)

if __name__ == '__main__':
    main()

运行结果：

(1, 2, 3)
6
tensor([[[1, 2, 3],
         [4, 5, 6]]], dtype=torch.int32)
torch.Size([1, 2, 3])
tensor([[[1, 4]],

        [[2, 5]],

        [[3, 6]]], dtype=torch.int32)
torch.Size([3, 1, 2])
tensor([[1, 4],
        [2, 5],
        [3, 6]], dtype=torch.int32)
torch.Size([3, 2])

标签：__,tensor,contiguous,torch,PyTorch,print,myArr
来源： https://blog.csdn.net/bigFatCat_Tom/article/details/97170387

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。