How to transpose 3D matrix in PyTorch v0.2

3

You are right. Thank you.