diff --git a/mindspore/common/initializer.py b/mindspore/common/initializer.py index c1e4cfc9db1..0e79763d4d3 100644 --- a/mindspore/common/initializer.py +++ b/mindspore/common/initializer.py @@ -272,6 +272,7 @@ class HeUniform(Initializer): .. math:: boundary = \sqrt{\frac{6}{(1 + a^2) \times \text{fan_in}}} + which is the bound of the HeUniform distribution. For details of HeUniform algorithm, please check