I would like to flatten a tensor into column-major order. What is the optimal way for achieving this in PyTorch? Row-major order seems to be the default in PyTorch’s flatten function, and I don’t think there is an order option like in Numpy’s flatten function. Thanks!