ApplyAlongAxis ============== Apply a function to 1D array slices along an axis. **Abstract Signature:** ``ApplyAlongAxis(func1d: Callable, axis: int, arr: Tensor, args, kwargs)`` .. raw:: html
jax.numpy.apply_along_axis
numpy.apply_along_axis