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:源維數,想要進行交
相關文章
相關標籤/搜索