diff --git a/mindspore/nn/layer/math.py b/mindspore/nn/layer/math.py index acf9a71190..878a2c1f42 100644 --- a/mindspore/nn/layer/math.py +++ b/mindspore/nn/layer/math.py @@ -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.