tf.experimental.numpy.take

Вариант TensorFlow функции NumPy's take.

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

Аргумент out не поддерживается, и по умолчанию используется режим clip.

См. документацию NumPy для numpy.take.