2019-03-04 20:12:22 -05:00
|
|
|
|
|
|
|
|
jax.ops package
|
|
|
|
|
=================
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: jax.ops
|
|
|
|
|
|
|
|
|
|
.. automodule:: jax.ops
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Indexed update operators
|
|
|
|
|
------------------------
|
|
|
|
|
|
|
|
|
|
JAX is intended to be used with a functional style of programming, and hence
|
2019-07-20 14:40:31 +01:00
|
|
|
does not support NumPy-style indexed assignment directly. Instead, JAX provides
|
|
|
|
|
pure alternatives, namely :func:`jax.ops.index_update` and its relatives.
|
2019-03-04 20:12:22 -05:00
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
|
|
index
|
|
|
|
|
index_update
|
|
|
|
|
index_add
|
2019-06-23 15:32:43 -07:00
|
|
|
index_min
|
|
|
|
|
index_max
|
2019-07-20 14:40:31 +01:00
|
|
|
|
|
|
|
|
Other operators
|
|
|
|
|
---------------
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
2019-05-08 21:18:17 -04:00
|
|
|
segment_sum
|
