jax.nn.initializers.lecun_normal#

jax.nn.initializers.lecun_normal(in_axis=-2, out_axis=-1, batch_axis=(), dtype=<class 'jax.numpy.float64'>)[源代码]#

构建一个 Lecun 正态初始化器。

Lecun 正态初始化器是 jax.nn.initializers.variance_scaling() 的一种特殊情况，其中 scale = 1.0，mode="fan_in"，并且 distribution="truncated_normal"。

参数：

  • in_axis (int | Sequence[int]) – 权重数组中输入维度的轴或轴序列。

  • out_axis (int | Sequence[int]) – 权重数组中输出维度的轴或轴序列。

  • batch_axis (Sequence[int]) – 权重数组中应忽略的轴或轴序列。

  • dtype (DTypeLikeInexact) – 权重的 dtype。

返回：

一个初始化器。

返回类型：

Initializer

示例

>>> import jax, jax.numpy as jnp
>>> initializer = jax.nn.initializers.lecun_normal()
>>> initializer(jax.random.key(42), (2, 3), jnp.float32)  
Array([[ 0.46700746,  0.8414632 ,  0.8518669 ],
       [-0.61677957, -0.67402434,  0.09683388]], dtype=float32)