pytorch中的transpose用法

注意：维数从0开始，0维 1维2维…,负数代表从右往左数，-1代表第一维，以此类推

import torch import numpy as np # 创建一个二维数组 arr = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]]) print("原始数组：") print(arr) # torch.transpose转置数组, dim的顺序谁先谁后没有关系，不影响最后结果！！！ # transpose 调换；移项；颠倒顺序 transposed_arr = arr.transpose(0, 1) print("转置后的数组：") print(transposed_arr) print("-----------------------------------------------------") transposed_arr = arr.transpose(-1, -2) print("转置后的数组：") print(transposed_arr)