mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 21:46:05 +00:00
f51a05a889
These functions have been deprecated and have issued a DeprecationWarning since jax 0.2.22 in October 2021.
25 lines
484 B
ReStructuredText
25 lines
484 B
ReStructuredText
|
|
jax.ops package
|
|
===============
|
|
|
|
.. 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
|