标签：Tensor reshape torch transpose shape PyTorch 方法 view

import torch

文章目录

• 1、reshape方法
• 2、view方法
• 3、permute方法
• 4、transpose方法
• 5、T属性

1、reshape方法

用法与NumPy的一模一样，既可以从torch这个库调用，torch.reshape(input, shape)，也可以在Tensor对象上调用torch.Tensor.reshape(shape)。

2、view方法

与reshape方法相比有较大的局限性，原因在于

1. 只能在Tensor对象上调用
2. Tensor对象必须是contiguous的，故常与contiguous方法连用：torch.Tensor.contiguous().view(size)，效果等价于torch.Tensor.reshape(shape)

所以，用view方法还不如直接用reshape方便

3、permute方法

permute方法是基于Tensor对象的，作用是置换Tensor的某几个维度。

4、transpose方法

基于torch库或Tensor对象，作用是置换Tensor的某两个维度。

5、T属性

返回原张量的转置，其对应的shape逆序。

>>>a = torch.randn(2,3,4,5)
>>>a.size()
torch.Size([2, 3, 4, 5])
>>>a.T.size()
torch.Size([5, 4, 3, 2])

标签：Tensor,reshape,torch,transpose,shape,PyTorch,方法,view
来源： https://blog.csdn.net/jasminefeng/article/details/113827268

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