Pytorch：permute（）函数

最新推荐文章于 2024-08-29 08:23:42 发布

北京地铁1号线

最新推荐文章于 2024-08-29 08:23:42 发布

阅读量733

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_54708219/article/details/129325046

版权

文章介绍了在PyTorch中如何使用`.permute()`函数来交换张量的维度，以一个3维张量`a`为例，展示了将维度索引1和2交换后的结果`b`，并解释了尺寸变化的过程。交换后张量`b`的形状从`(1,2,3,4)`变为`(4,1,3,2)`。

摘要由CSDN通过智能技术生成

输入：维度索引（所有）；

输出：交换维度之后的张量。

功能：可以同时多次交换Tensor的维度

举例：b = a.permute(0, 2 ,1)表示，a是一个3维的Tensor，将a的维度索引1和维度索引2交换位置，而维度索引0表示不变。

import torch
a=torch.rand(1,2,3,4)
print(a)
b=a.permute(3,0,2,1)
print(b)
print(b.shape)

输出：

tensor([[[[0.0564, 0.3929, 0.3702, 0.3258],
          [0.3509, 0.8068, 0.8201, 0.1457],
          [0.1383, 0.2084, 0.6474, 0.3917]],

         [[0.8784, 0.3841, 0.7031, 0.3566],
          [0.0505, 0.1398, 0.8103, 0.3775],
          [0.5874, 0.8171, 0.4946, 0.4990]]]])
tensor([[[[0.0564, 0.8784],
          [0.3509, 0.0505],
          [0.1383, 0.5874]]],


        [[[0.3929, 0.3841],
          [0.8068, 0.1398],
          [0.2084, 0.8171]]],


        [[[0.3702, 0.7031],
          [0.8201, 0.8103],
          [0.6474, 0.4946]]],


        [[[0.3258, 0.3566],
          [0.1457, 0.3775],
          [0.3917, 0.4990]]]])
torch.Size([4, 1, 3, 2])

这里张量a的size为（1，2，3，4），经过b=a.permute(3,0,2,1)之后变成了size(4，1，3，2)。