Announcing the TensorFlow Dev Summit 2020 Learn more

tf.batch_gather

View source on GitHub

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

Aliases:

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