Chose rows of 3d Tensor based on some repeated indices. Tricky Slicing

Question

So this is a tricky bit of Tensor slicing I\'m trying.

I have a tensor A which is 3d

>> A.shape
torch.Size([60, 10, 16])