jax.nn.initializers.constant

jax.nn.initializers.constant(value, dtype=<class 'jax.numpy.float64'>)

Builds an initializer that returns arrays full of a constant value.

Parameters:

• value (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – the constant value with which to fill the initializer.

• dtype (Any) – optional; the initializer’s default dtype.

>>> import jax, jax.numpy as jnp
>>> initializer = jax.nn.initializers.constant(-7)
:rtype: :sphinx_autodoc_typehints_type:`\:py\:class\:\`\~jax.nn.initializers.Initializer\``
>>> initializer(jax.random.key(42), (2, 3), jnp.float32)
Array([[-7., -7., -7.],
       [-7., -7., -7.]], dtype=float32)