tensor-indexing

Pytorch: How to reorder the tensor by given sorted indices

Pytorch: How to reorder the tensor by given sorted indices Question: Given a tensor A shape (d0, d1, …, dn, dn+1) and a tensor of sorted indices I with shape (d0, d1, …, dn) I want to reorder the indices of A using the sorted indices in I. The first n dimensions of tensors A …

Total answers: 1