Pytorch学习之常用函数

笃静悟初

于 2019-05-27 20:32:24 发布

阅读量120

点赞数

分类专栏： pytorch学习文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/innocent_cat/article/details/90611257

版权

pytorch学习专栏收录该内容

7 篇文章

订阅专栏

博客介绍了Tensor.permute()和Tensor.transpose()的作用，二者均可用于高维转置。其中，permute可对任意高维矩阵转置，且只能通过Tensor.permute()调用；而transpose只能操作2D矩阵转置，有两种调用方式，连续使用也能实现permute的效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Tensor.permute()
作用：高维转置
permute可以对任意高维矩阵进行转置.
但没有 torch.permute() 这个调用方式，只能 Tensor.permute()。
```
  torch.rand(2,3,4,5).permute(3,2,0,1).shape	
  
  Out: torch.Size([5, 4, 2, 3])
```

Tensor.transpose()
作用：高维转置
transpose只能操作2D矩阵的转置。有两种调用方式。连续使用transpose也可实现permute的效果。

 torch.transpose(Tensor, 1, 0)

 t.rand(2,3,4,5).transpose(3,0).transpose(2,1).transpose(3,2).shape	
 Out: torch.Size([5, 4, 2, 3])

 t.rand(2,3,4,5).transpose(1,0).transpose(2,1).transpose(3,1).shape
 Out: torch.Size([3, 5, 2, 4])