sciai.common.LeCunUniform

class sciai.common.LeCunUniform[源代码]

生成一个服从Yan LeCun均匀分布 \({N}(-\text{boundary}, \text{boundary})\) 的随机数组用于初始化Tensor，其中：

\[boundary = \sqrt{\frac{3}{fan\_in}}\]

fan_in 是权重Tensor中输入单元的数量。

更多关于Yan LeCun均匀分布的细节请参考： Neural Tangent Kernel: Convergence and Generalization in Neural Networks。

支持平台：: GPU CPU Ascend

样例：

>>> import mindspore
>>> from mindspore.common.initializer import initializer
>>> from sciai.common.initializer import LeCunUniform
>>> tensor = initializer(LeCunUniform(), [1, 2, 3], mindspore.float32)