jax.nn.initializers.Initializer

class jax.nn.initializers.Initializer(*args, **kwargs)

Protocol for initializers returned by jax.nn.initializers APIs.

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)