mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
24 lines
489 B
ReStructuredText
24 lines
489 B
ReStructuredText
``jax.ops`` module
|
|
==================
|
|
|
|
.. currentmodule:: jax.ops
|
|
|
|
.. automodule:: jax.ops
|
|
|
|
.. _syntactic-sugar-for-ops:
|
|
|
|
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.
|
|
|
|
Segment reduction operators
|
|
---------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
segment_max
|
|
segment_min
|
|
segment_prod
|
|
segment_sum
|