I don’t quite understand the operation you are trying to achive.
Are you looking for reshaping the tensor? or creating a tensor based on a selection from a tensor of indexes? Also, where the zero came from on the resulting tensor tensor([[4, 3], [6,0]])?