Swap axes in pytorch?

3

Thanks, I have just checked the Docs, but it seems that I just miss it…