pytorch实现HWC转CHW代码示例

本篇文章小编给大家分享一下pytorch实现HWC转CHW代码示例，文章代码介绍的很详细，小编觉得挺不错的，现在分享给大家供大家参考，有需要的小伙伴们可以来看看。

代码如下：

import torch
import numpy as np
from torchvision.transforms import ToTensor 
t = torch.tensor(np.arange(24).reshape(2,4,3))
print(t)
 
#HWC 转CHW
print(t.transpose(0,2).transpose(1,2)) 
print(t.permute(2,0,1)) 
print(ToTensor()(t.numpy()))

D:anacondapython.exe C:/Users/liuxinyu/Desktop/pytorch_test/day3/hwc转chw.py

tensor([[[ 0, 1, 2],

[ 3, 4, 5],

[ 6, 7, 8],

[ 9, 10, 11]],

[[12, 13, 14],

[15, 16, 17],

[18, 19, 20],

[21, 22, 23]]], dtype=torch.int32)

tensor([[[ 0, 3, 6, 9],

[12, 15, 18, 21]],

[[ 1, 4, 7, 10],

[13, 16, 19, 22]],

[[ 2, 5, 8, 11],

[14, 17, 20, 23]]], dtype=torch.int32)

tensor([[[ 0, 3, 6, 9],

[12, 15, 18, 21]],

[[ 1, 4, 7, 10],

[13, 16, 19, 22]],

[[ 2, 5, 8, 11],

[14, 17, 20, 23]]], dtype=torch.int32)

tensor([[[ 0, 3, 6, 9],

[12, 15, 18, 21]],

[[ 1, 4, 7, 10],

[13, 16, 19, 22]],

[[ 2, 5, 8, 11],

[14, 17, 20, 23]]], dtype=torch.int32)

Process finished with exit code 0