From 04c34579b766bbf4bc1e910acdca451251f12a11 Mon Sep 17 00:00:00 2001
From: peixu_ren <peixu.ren1@huawei.com>
Date: Wed, 23 Dec 2020 09:54:54 -0500
Subject: [PATCH] Supplement the formula of ReduceLogSumExp

---
 mindspore/nn/layer/math.py | 4 ++++
 1 file changed, 4 insertions(+)

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.