sciai.common.LeCunNormal

查看源文件
class sciai.common.LeCunNormal[源代码]

生成一个服从Yan LeCun正态分布 \({N}(0, \text{sigma}^2)\) 的随机数组用于初始化Tensor,其中:

\[sigma = \sqrt{\frac{1}{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 LeCunNormal
>>> tensor = initializer(LeCunNormal(), [1, 2, 3], mindspore.float32)