0
点赞
收藏
分享

微信扫一扫

torch.transpose

Java旺 2022-02-20 阅读 149

torch.transpose
作用:一个tensor的两维度进行交换,只能有两个相互交换的位置参数

>>> x = torch.randn((2, 3,4))
>>> x.size()
torch.Size([2, 3, 4])

>>> y = torch.transpose(x,1,2) # 交换第二维和第三维
>>> y.size()
torch.Size([2, 4, 3])

>>> y = torch.transpose(x,0,1) # 交换第一维和第二维
>>> y.size()
torch.Size([3, 2, 4])
举报

相关推荐

0 条评论