mirror of
https://github.com/ROCm/jax.git
synced 2025-04-25 01:06:05 +00:00
30 lines
488 B
ReStructuredText
30 lines
488 B
ReStructuredText
|
|
jax.nn.initializers package
|
|
===========================
|
|
|
|
.. currentmodule:: jax.nn.initializers
|
|
|
|
.. automodule:: jax.nn.initializers
|
|
|
|
|
|
Initializers
|
|
------------
|
|
|
|
This module provides common neural network layer initializers,
|
|
consistent with definitions used in Keras and Sonnet.
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
zeros
|
|
ones
|
|
uniform
|
|
normal
|
|
variance_scaling
|
|
glorot_uniform
|
|
glorot_normal
|
|
lecun_uniform
|
|
lecun_normal
|
|
he_uniform
|
|
he_normal
|