pytorch学习随笔记录二——遇到的未知函数记录

numpy数组维度互换函数swapaxes

由于本次写的程序需要将数据数组除了batch之外，每次需要二次分组。所以处理数据维度又多了一层，但是经常需要变换两个分组的维度。此处用到swapaxes函数。数组是numpy库的。

num = np.random.random((2,3,4))
print(num)
num = np.swapaxes(num,0,2)
print(num)

# [[[0.0372153  0.56444485 0.30521904 0.10190487]
#   [0.88590199 0.05494382 0.92141733 0.45761658]
#  [0.67246559 0.20745557 0.24289462 0.52821502]]
# 
# [[0.91978977 0.00412304 0.58665158 0.53105978]
#  [0.59273119 0.09831343 0.07644877 0.12664533]
# [0.24492794 0.59717912 0.47824424 0.19823922]]]
# [[[0.0372153  0.91978977]
#   [0.88590199 0.59273119]
#  [0.67246559 0.24492794]]
# 
# [[0.56444485 0.00412304]
#  [0.05494382 0.09831343]
# [0.20745557 0.59717912]]
# 
# [[0.30521904 0.58665158]
#  [0.92141733 0.07644877]
# [0.24289462 0.47824424]]
# 
# [[0.10190487 0.53105978]
#  [0.45761658 0.12664533]
# [0.52821502 0.19823922]]]

从历程可以看出，函数可以实现数组或者矩阵任意两个维度互换。