Browse Source

fix HSigmoid comment

tags/v0.7.0-beta
wangdongxu 5 years ago
parent
commit
b59e5b60fe
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/nn/layer/activation.py
  2. +1
    -1
      mindspore/ops/operations/nn_ops.py

+ 1
- 1
mindspore/nn/layer/activation.py View File

@@ -454,7 +454,7 @@ class HSigmoid(Cell):
Hard sigmoid is defined as:

.. math::
\text{hsigmoid}(x_{i}) = max(0, min(1, \frac{2 * x_{i} + 5}{10})),
\text{hsigmoid}(x_{i}) = max(0, min(1, \frac{x_{i} + 3}{6})),

where :math:`x_{i}` is the :math:`i`-th slice along the given dim of the input Tensor.



+ 1
- 1
mindspore/ops/operations/nn_ops.py View File

@@ -498,7 +498,7 @@ class HSigmoid(PrimitiveWithInfer):
Hard sigmoid is defined as:

.. math::
\text{hsigmoid}(x_{i}) = max(0, min(1, \frac{2 * x_{i} + 5}{10})),
\text{hsigmoid}(x_{i}) = max(0, min(1, \frac{x_{i} + 3}{6})),

where :math:`x_{i}` is the :math:`i`-th slice along the given dim of the input Tensor.



Loading…
Cancel
Save