https://jdhao.github.io/2019/07/10/pytorch_view_reshape_transpose_permute/