I was reimplementing a tensorflow code in pytorch but found that there is no corresponding functions for tf.batch_gather, like the following code,
new_neighbors_indices = tf.batch_gather(neighbors_indices, inds)

The dimension of neighbors_indices is [a, b] and the dimension of inds is [a, c]. Is there any ways to implement such operations in pytorch?