Hi,

I am trying to permute the feature map dimension in a tensor. As a very simplified case,

If I have a tensor of size (5, 4, 3, 6)

I want to rearrange the above tensor along its dimension 1 (i.e. 4) from 0,1,2,3 to 0,2,1,3

One possible way I found was to do a index_select followed by cat. But, in case of a larger tensor, a lot of intermediate tensors will have to be created.

Is there a better way to achieve this?

Regards

Nabarun