jax.nn.initializers.normal#

jax.nn.initializers.normal(stddev=0.01, dtype=<class 'jax.numpy.float64'>)[原始碼]#

建立一個初始化器，其返回實數常態分佈的隨機陣列。

參數:

  • stddev (RealNumeric) – 選填；分佈的標準差。

  • dtype (DTypeLikeInexact) – 選填；初始化器的預設 dtype。

返回:

一個初始化器，其返回的陣列的值呈常態分佈，平均值為 0，標準差為 stddev。

返回型別:

Initializer

>>> 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)