mlx.nn.init.constant#

constant(value: float, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#

一个初始化器，返回一个填充了 value 的数组。

参数:

• value (float) – 用于填充数组的值。

• dtype (Dtype, optional) – 数组的数据类型。默认值：float32。

返回:

一个初始化器，返回一个形状与输入相同，并填充了 value 的数组。

返回类型:

Callable[[array], array]

示例

>>> init_fn = nn.init.constant(0.5)
>>> init_fn(mx.zeros((2, 2)))
array([[0.5, 0.5],
       [0.5, 0.5]], dtype=float32)