Browse Source

Supplement the formula of ReduceLogSumExp

tags/v1.1.0
peixu_ren 5 years ago
parent
commit
04c34579b7
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/nn/layer/math.py

+ 4
- 0
mindspore/nn/layer/math.py View File

@@ -50,6 +50,10 @@ class ReduceLogSumExp(Cell):

The dtype of the tensor to be reduced is number.

.. math::

ReduceLogSumExp(x) = \log(\sum(e^x))

Args:
axis (Union[int, tuple(int), list(int)]) - The dimensions to reduce. Default: (), reduce all dimensions.
Only constant value is allowed.


Loading…
Cancel
Save