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