jax.ops module
Contents
jax.ops module#
The functionsjax.ops.index_update,jax.ops.index_add, etc., which weredeprecated in JAX 0.2.22, have been removed. Please use thejax.numpy.ndarray.at property on JAX arrays instead.
Segment reduction operators#
| Computes the maximum within segments of an array. |
| Computes the minimum within segments of an array. |
| Computes the product within segments of an array. |
| Computes the sum within segments of an array. |
Contents