Let a be a (n, d, l) tensor. Let indices be a (n, 1) tensor, containing indices. I want to gather from a in
a
(n, d, l)
indices
(n, 1)
