Как поменять местами ось тензора резака?

Question

У меня есть тензор факела размером torch.Size([1, 128, 56, 128])

1 - канал, 128 - ширина и высота. 56 - стопки изображений.

Как я могу изменить его размер до torch.Size([1, 56, 128, 128])?

thankyou

Answer 1

вы можете просто использовать permute or transpose