jax.nn.initializers.normal#

jax.nn.initializers.normal(stddev=0.01, dtype=<class 'jax.numpy.float64'>)[源代码]#

构建一个返回实数正态分布随机数组的初始化器。

参数：

  • stddev (实数数值) – 可选；分布的标准差。

  • dtype (非精确的 DTypeLike) – 可选；初始化器的默认数据类型。

返回：

一个初始化器，它返回均值为 0，标准差为 stddev 的正态分布数组。

返回类型：

初始化器

>>> import jax, jax.numpy as jnp
>>> initializer = jax.nn.initializers.normal(5.0)
>>> initializer(jax.random.key(42), (2, 3), jnp.float32)  
Array([[ 3.0613258 ,  5.6129413 ,  5.6866574 ],
       [-4.063663  , -4.4520254 ,  0.63115686]], dtype=float32)