mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 12:36:07 +00:00
25 lines
405 B
ReStructuredText
25 lines
405 B
ReStructuredText
|
|
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
|
|
does not support NumPy-style indexed assignment directly.
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
index
|
|
index_update
|
|
index_add
|
|
index_min
|
|
index_max
|
|
segment_sum
|