refine logsumexp error message and docs

PaddlePaddle · Sep 29, 2020 · 638c2db · 638c2db · paddle-bot-old · Sep 30, 2020
1 parent 7b46fb0
commit 638c2db
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 18 deletions.
diff --git a/paddle/fluid/operators/reduce_ops/logsumexp_op.cc b/paddle/fluid/operators/reduce_ops/logsumexp_op.cc
@@ -32,7 +32,7 @@ class LogsumexpOp : public framework::OperatorWithKernel {
     PADDLE_ENFORCE_LE(x_rank, 4,
                       platform::errors::InvalidArgument(
                           "The input tensor X's dimensions of logsumexp "
-                          "should be less equal than 4. But received X's "
+                          "should be less or equal than 4. But received X's "
                           "dimensions = %d, X's shape = [%s].",
                           x_rank, x_dims));
     auto axis = ctx->Attrs().Get<std::vector<int>>("axis");
@@ -45,20 +45,18 @@ class LogsumexpOp : public framework::OperatorWithKernel {
             axis.size()));
 
     for (size_t i = 0; i < axis.size(); i++) {
-      PADDLE_ENFORCE_LT(
-          axis[i], x_rank,
-          platform::errors::InvalidArgument(
-              "axis[%d] should be in the "
-              "range [-dimension(X), dimension(X)] "
-              "where dimesion(X) is %d. But received axis[i] = %d.",
-              i, x_rank, axis[i]));
-      PADDLE_ENFORCE_GE(
-          axis[i], -x_rank,
-          platform::errors::InvalidArgument(
-              "axis[%d] should be in the "
-              "range [-dimension(X), dimension(X)] "
-              "where dimesion(X) is %d. But received axis[i] = %d.",
-              i, x_rank, axis[i]));
+      PADDLE_ENFORCE_LT(axis[i], x_rank,
+                        platform::errors::InvalidArgument(
+                            "axis[%d] should be in the "
+                            "range [-D, D), where D is the dimensions of X and "
+                            "D is %d. But received axis[%d] = %d.",
+                            i, x_rank, i, axis[i]));
+      PADDLE_ENFORCE_GE(axis[i], -x_rank,
+                        platform::errors::InvalidArgument(
+                            "axis[%d] should be in the "
+                            "range [-D, D), where D is the dimensions of X and "
+                            "D is %d. But received axis[%d] = %d.",
+                            i, x_rank, i, axis[i]));
       if (axis[i] < 0) {
         axis[i] += x_rank;
       }

diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py
@@ -1006,7 +1006,7 @@ def logsumexp(x, axis=None, keepdim=False, name=None):
     This OP calculates the log of the sum of exponentials of ``x`` along ``axis`` .
 
     .. math::
-       logsumexp(x) = \log\sum exp(x)
+       logsumexp(x) = \\log\\sum exp(x)
 
     Args:
         x (Tensor): The input Tensor with data type float32, float64.
@@ -1037,8 +1037,6 @@ def logsumexp(x, axis=None, keepdim=False, name=None):
 
         import paddle
 
-        paddle.disable_static()
-
         x = paddle.to_tensor([[-1.5, 0., 2.], [3., 1.2, -2.4]])
         out1 = paddle.logsumexp(x) # [3.4691226]
         out2 = paddle.logsumexp(x, 1) # [2.15317821, 3.15684602]