tf.experimental.numpy.take

TensorFlow variant of NumPy's take.

tf.experimental.numpy.take(
    a, indices, axis=None, out=None, mode='clip'
)

out argument is not supported, and default mode is clip.

See the NumPy documentation for numpy.take.