|---| | [ View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.21.0/tensorflow/python/ops/numpy_ops/np_array_ops.py#L1607-L1668) | TensorFlow variant of NumPy's `take_along_axis`. tf.experimental.numpy.take_along_axis( arr, indices, axis ) See the NumPy documentation for [`numpy.take_along_axis`](https://numpy.org/doc/stable/reference/generated/numpy.take_along_axis.html).