mlx.nn.init.uniform

uniform(low: float = 0.0, high: float = 1.0, dtype: Dtype = mlx.core.float32) → Callable[[array], array]

一个返回均匀分布样本的初始化器。

参数：

• low (float, optional) – 均匀分布的下界。默认值：0.0。

• high (float, optional) – 均匀分布的上界。默认值：1.0

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

返回值：

一个初始化器，返回一个与输入具有相同形状且填充有均匀分布样本的数组

返回类型：

Callable[[array], array]

示例

>>> init_fn = nn.init.uniform(low=0, high=1)
>>> init_fn(mx.zeros((2, 2)))
array([[0.883935, 0.863726],
       [0.617261, 0.417497]], dtype=float32)