【torch】torch中的permute用法

最新推荐文章于 2024-07-25 19:48:34 发布

zkq_1986

最新推荐文章于 2024-07-25 19:48:34 发布

阅读量1.6k

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/zkq_1986/article/details/107579489

版权

pytorch 专栏收录该内容

5 篇文章 2 订阅

订阅专栏

permute将tensor的维度进行转换。

举例：

import torch
import numpy as np

a=np.array([[[1,2,3],[4,5,6]]])

b=torch.tensor(a)
print(b)
print(b.size()) #  ——>  torch.Size([1, 2, 3])

c=b.permute(2,0,1)
print(c.size()) #  ——>  torch.Size([3, 1, 2])

print(c)

c = b.view(3,1,2)
print(c)

输出：

tensor([[[1, 2, 3],
[4, 5, 6]]], dtype=torch.int32)
torch.Size([1, 2, 3])
torch.Size([3, 1, 2])
tensor([[[1, 4]],

[[2, 5]],

[[3, 6]]], dtype=torch.int32)
tensor([[[1, 2]],

[[3, 4]],

[[5, 6]]], dtype=torch.int32)