jax.ops
module#
The functions jax.ops.index_update
, jax.ops.index_add
, etc., which were
deprecated in JAX 0.2.22, have been removed. Please use the
jax.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. |