tf.keras.ops.take_along_axis
Select values from x at the 1-D indices along the given axis.
tf.keras.ops.take_along_axis(
x, indices, axis=None
)
| Args | |
|---|---|
x | Source tensor. |
indices | The indices of the values to extract. |
axis | The axis over which to select values. By default, the flattened input tensor is used. |
| Returns | |
|---|---|
| The corresponding tensor of values. |
© 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
