gistfile1.txt

import torch
self = torch.rand((10, 1))
print(self.stride())
order =[i[0] for i in sorted(
    enumerate(self.stride()), key=lambda x: x[1], reverse=True
)]
print(order)
self_permuted = self.permute((1, 0))
print(self_permuted.stride())
order =[i[0] for i in sorted(
    enumerate(self_permuted.stride()), key=lambda x: x[1], reverse=True
)]
print(order)