Apparently, it does, thanks.
@eqy would torch.gather safe for keeping the gradient?
I see in the torch.gather documentations 4 that it has something called sparse_grad, I did not quite understand what does it mean to have gradient to be a sparse tensor.…