jax.nn.initializers.glorot_uniform

jax.nn.initializers.glorot_uniform(in_axis=-2, out_axis=-1, batch_axis=(), dtype=<class 'jax.numpy.float64'>)

构建 Glorot 均匀初始化器（又名 Xavier 均匀初始化器）。

Glorot 均匀初始化器 是 jax.nn.initializers.variance_scaling() 的特化，其中 scale = 1.0, mode="fan_avg", 和 distribution="uniform"。

参数:

• in_axis (返回:

  一个初始化器。

  返回类型:

  Initializer

示例

>>> import jax, jax.numpy as jnp
>>> initializer = jax.nn.initializers.glorot_uniform()
>>> initializer(jax.random.key(42), (2, 3), jnp.float32)  
Array([[ 0.50350785,  0.8088631 ,  0.81566876],
       [-0.6393332 , -0.6865721 ,  0.11003882]], dtype=float32)