Ask for help! I want to know a best practice about torch.Tensor

2

I believe that is done by:

A.gather(-2, I.view(256,960*2,1).expand(-1,-1,1024)).view(256,960,2,1024)

1 Like