So I have this tensor t of size a x b x c
I have another tensor u of size a. Each element of u is an index of the element I want from t’s second dimension. So I want to apply the mask and get something that’s a x c.
How can I do this? Please let me know if I can provide any clarifications.