ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.experimental.numpy.take_along_axis

TensorFlow variant of NumPy's take_along_axis.

See the NumPy documentation for numpy.take_along_axis.