pytorch中的transpose()

import torch
batch_index = torch.randn(2, 3, 4)
batch_es = batch_index.transpose(0, 1)
batch_ee = batch_index.transpose(1, 0)
print(batch_index)
print("&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&")
print(batch_es)
print("*********************************")
print(batch_ee)

运行结果：

batch_es和batch_ee输出的结果是一样的。原始的transpose参数(默认的参数)为(0,1,2)，这个转置相当于将第一个坐标与第二坐标进行了互换。