Torch和Numpy——形状变换与轴交换

输入

import numpy as np
import torch

#numpy创建
a = np.arange(1,12,2).reshape([2,3]) #生成一个由1到12之间的数组成的步长为2的2x3矩阵
print(a)
b=a.T  #对a进行转置
print(b)
print('&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&')

c = np.arange(12).reshape([2,2,3])  #生成一个由0到11的数字组成的形状为2组2x3的矩阵
print(c)
print("_________________________________________________________________")

#torch创建
a1 = torch.arange(12).reshape([3,4])
print(a1)
b1 = a1.T  #或者  b1 = a1.t()
print(b1)
print("**********************************************************************")


c1 = torch.arange(12).reshape([2,2,3])  #生成一个有0到11的数字组成的形状为2组2x3的矩阵
print(c1)

d = torch.transpose(c1,0,2)  #将c1的0轴和2轴进行交换
print(d)
d1 = torch.transpose(d,1,2)
print(d1)
print("^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^")

d2 = c1.permute([2,0,1])  #第二种交换c1的3个轴的方法
print(d2)

e = np.reshape(c,[3,2,2])  #矩阵变形
print(e)

输出

[[ 1  3  5]
 [ 7  9 11]]
[[ 1  7]
 [ 3  9]
 [ 5 11]]
&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
[[[ 0  1  2]
  [ 3  4  5]]

 [[ 6  7  8]
  [ 9 10 11]]]
_________________________________________________________________
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
tensor([[ 0,  4,  8],
        [ 1,  5,  9],
        [ 2,  6, 10],
        [ 3,  7, 11]])
**********************************************************************
tensor([[[ 0,  1,  2],
         [ 3,  4,  5]],

        [[ 6,  7,  8],
         [ 9, 10, 11]]])
tensor([[[ 0,  6],
         [ 3,  9]],

        [[ 1,  7],
         [ 4, 10]],

        [[ 2,  8],
         [ 5, 11]]])
tensor([[[ 0,  3],
         [ 6,  9]],

        [[ 1,  4],
         [ 7, 10]],

        [[ 2,  5],
         [ 8, 11]]])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
tensor([[[ 0,  3],
         [ 6,  9]],

        [[ 1,  4],
         [ 7, 10]],

        [[ 2,  5],
         [ 8, 11]]])
[[[ 0  1]
  [ 2  3]]

 [[ 4  5]
  [ 6  7]]

 [[ 8  9]
  [10 11]]]