sciai.common.LeCunUniform

View Source On Gitee
class sciai.common.LeCunUniform[source]

Yann LeCun Normal Initialization \({U}(-\text{boundary}, \text{boundary})\) in order to initialize a tensor, where

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

‘fan_in’ is the number of input units of the weight tensor.

For details of LeCun Uniform Initialization, please check: Neural Tangent Kernel: Convergence and Generalization in Neural Networks.

Supported Platforms:

GPU CPU Ascend

Examples

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