mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
22 lines
261 B
ReStructuredText
22 lines
261 B
ReStructuredText
``jax.tree`` module
|
|
===================
|
|
|
|
.. currentmodule:: jax.tree
|
|
|
|
.. automodule:: jax.tree
|
|
|
|
List of Functions
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
all
|
|
flatten
|
|
leaves
|
|
map
|
|
reduce
|
|
structure
|
|
transpose
|
|
unflatten
|