tf.batch_gather

tf.batch_gather(
    params,
    indices,
    name=None
)

Defined in tensorflow/python/ops/array_ops.py.

Gather slices from params according to indices with leading batch dims. (deprecated)