tf.keras.ops.take_along_axis
Выбирает значения из x по одномерному indices вдоль указанной оси.
tf.keras.ops.take_along_axis(
x, indices, axis=None
)
| Аргументы | |
|---|---|
x | Исходный тензор. |
indices | Индексы значений, которые нужно извлечь. |
axis | Ось, вдоль которой выбираются значения. По умолчанию используется входной тензор, преобразованный в одномерный массив. |
| Возвращает | |
|---|---|
| Соответствующий тензор значений. |
© 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/api_docs/python/tf/keras/ops/take_along_axis
