jax.nn.initializers.zeros

jax.nn.initializers.zeros(key, shape, dtype=<class 'jax.numpy.float64'>)

一個初始化器，返回一個充滿零的常數陣列。

key 參數會被忽略。

>>> import jax, jax.numpy as jnp
>>> jax.nn.initializers.zeros(jax.random.key(42), (2, 3), jnp.float32)
Array([[0., 0., 0.],
       [0., 0., 0.]], dtype=float32)

參數:

• key (Array)

• shape (core.Shape)

• dtype (DTypeLikeInexact)

返回類型:

Array