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)