torch.transpose

Java旺

关注

阅读 149

2022-02-20

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)

0 0 举报