Supose i have a tensor A of size  batch_size x num_class x Dim, and a batch of labels  L of size batch_size, where each element specifies which number in the second dim to choose from.  And i want to slice tensor A using L, resulting a tensor: bz x Dim,  my question is, is there a good way to implement this.  can anyone help me out?