pytorch
x = torch.transpose(x, 1, 2)
交换1和2维度,维度可以多个但,只能两个维度交换。
x = torch.permute(1, 2, 0)
交换多个维度
numpy
x = numpy.transpose(1, 2, 0)
交换多个维度
x = torch.transpose(x, 1, 2)
交换1和2维度,维度可以多个但,只能两个维度交换。
x = torch.permute(1, 2, 0)
交换多个维度
x = numpy.transpose(1, 2, 0)
交换多个维度