2019-03-04 20:12:22 -05:00
|
|
|
|
|
|
|
|
jax.ops package
|
2022-02-17 15:23:44 -05:00
|
|
|
===============
|
2019-03-04 20:12:22 -05:00
|
|
|
|
|
|
|
|
.. currentmodule:: jax.ops
|
|
|
|
|
|
|
|
|
|
.. automodule:: jax.ops
|
|
|
|
|
|
2021-05-10 17:44:18 -04:00
|
|
|
.. _syntactic-sugar-for-ops:
|
2019-03-04 20:12:22 -05:00
|
|
|
|
2022-02-17 15:23:44 -05:00
|
|
|
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
|
|
|
|
|
:attr:`jax.numpy.ndarray.at` property on JAX arrays instead.
|
2019-03-04 20:12:22 -05:00
|
|
|
|
2022-02-17 15:23:44 -05:00
|
|
|
Segment reduction operators
|
|
|
|
|
---------------------------
|
2019-07-20 14:40:31 +01:00
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
2021-04-09 12:06:51 -07:00
|
|
|
segment_max
|
|
|
|
|
segment_min
|
|
|
|
|
segment_prod
|
2019-05-08 21:18:17 -04:00
|
|
|
segment_sum
|
