torch常用操作-torch.transpose、torch.permute、np.transpose

1. 简介

在使用Pytorch中我们经常会对矩阵进行操作,其中比较常用的就是调整维度,交换不同的轴。比如常见的有torch.transpose、torch.permute、np.transpose它们之间是有区别和联系的。

2. torch.transpose

首先举个例子,然后贴个官方文档,意思就是输入矩阵,然后交换两个维度(参数中只有两个维度)。
torch常用操作-torch.transpose、torch.permute、np.transpose_第1张图片
变换过程可以理解为:
torch常用操作-torch.transpose、torch.permute、np.transpose_第2张图片
torch常用操作-torch.transpose、torch.permute、np.transpose_第3张图片

3. torch.permute与np.transpose

这两个操作的效果其实是相同的都可以操作多个轴,只不过一个是tensor,一个是numpy数组。与torch.transpose的区别是permute可以变换多个轴,原理和上一小节一样直接交换坐标,看下官方解释:
torch常用操作-torch.transpose、torch.permute、np.transpose_第4张图片

4. 总结

以上的函数都可以进行轴转换操作。np.transpose 同 torch.permute一样,一次修改多个轴;torch.transpose一次只能交换2个轴。

5.参考链接

  1. https://pytorch.org/docs/stable/index.html
  2. https://blog.csdn.net/qq_28949847/article/details/128568827
  3. https://zhuanlan.zhihu.com/p/519296425

你可能感兴趣的:(Pytorch,python,深度学习)