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.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/experimental/numpy/take