灵巧戒指

文章
4
资源
0
加入时间
3年0月9天

torch.transpose()

首先transpose是为了转换矩阵维度的,在numpy和pytorch中的作用都是一样的,分别举例说明:arr = np.arange(24).reshape((2, 3, 4))[[[ 0 1 2 3] [ 4 5 6 7] [ 8 9 10 11]] [[12 13 14 15] [16 17 18 19] [20 21 22 23]]]目标:将矩阵 arr 的0维度和1维度进行交换需要注意,transpose每次只能转换两个维度np.transpose(