jax.nn.initializers.glorot_normal#
- jax.nn.initializers.glorot_normal(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="truncated_normal"
.- 参数:
in_axis (int | Sequence[int]) – 权重数组中输入维度的轴或轴序列。
out_axis (int | Sequence[int]) – 权重数组中输出维度的轴或轴序列。
batch_axis (Sequence[int]) – 权重数组中应忽略的轴或轴序列。
dtype (DTypeLikeInexact) – 权重的 dtype。
- 返回值:
一个初始化器。
- 返回类型:
初始化器
示例
>>> import jax, jax.numpy as jnp >>> initializer = jax.nn.initializers.glorot_normal() >>> initializer(jax.random.key(42), (2, 3), jnp.float32) Array([[ 0.41770416, 0.75262755, 0.7619329 ], [-0.5516644 , -0.6028657 , 0.08661086]], dtype=float32)