mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
Update docs/notebooks/JAX_pytrees.ipynb
Co-Authored-By: Stephan Hoyer <shoyer@google.com>
This commit is contained in:
parent
4e89d43a75
commit
b12a8019c8
@ -92,7 +92,7 @@
|
||||
"colab_type": "text"
|
||||
},
|
||||
"source": [
|
||||
"Pytrees containers can be lists, tuples, dicts, namedtuple. Leaves as numeric and ndarrays:"
|
||||
"Pytrees containers can be lists, tuples, dicts, namedtuple. Numeric and ndarray values are treated as leaves:"
|
||||
]
|
||||
},
|
||||
{
|
||||
@ -320,4 +320,4 @@
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user