pytorch每日一學36(torch.movedim())將tensor指定的維數進行交換

第36個方法 torch.movedim(input, source, destination) → Tensor 此方法是將input中source維與destination維進行交換，例如原來的維數是(3, 4, 5)通過此方法後能夠將維數改成(3, 5, 4)或者(4, 3, 5)均可以。 接下來介紹參數：python input：要交換維數的tensor。 source：源維數，想要進行交

>>阅读原文<<