jax.nn.initializers.lecun_uniform

jax.nn.initializers.lecun_uniform#

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

构建 Lecun 均匀初始化器。

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

参数:

  • 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.lecun_uniform()
>>> initializer(jax.random.key(42), (2, 3), jnp.float32)  
Array([[ 0.56293887,  0.90433645,  0.9119454 ],
       [-0.71479625, -0.7676109 ,  0.12302713]], dtype=float32)
内容