概述
- 函数作用
交换一个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.transpose()函数解读所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复