- 函数作用
交换一个tensor的两个维度
- 代码示例
>>> cc=torch.randn((2,3,4))
>>> dd=torch.transpose(cc,1,2)
>>> cc.size()
torch.Size([2, 3, 4])
>>> dd.size()
torch.Size([2, 4, 3])
>>> dd=torch.transpose(cc,0,2)
>>> dd.size()
torch.Size([4, 3, 2])
>>> dd=cc.transpose(1,2,0)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: transpose() takes 2 positional arguments but 3 were given
注意只能有两个相关的交换的位置参数。
最后
以上就是知性蓝天最近收集整理的关于torch.transpose()函数解读的全部内容，更多相关torch内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复