From 723cbe9ffa0a40ca2103c9023648f503469a0047 Mon Sep 17 00:00:00 2001 From: Shang Zhizhou Date: Tue, 7 Dec 2021 16:42:02 +0800 Subject: [PATCH] update logsumexp doc (#37883) * update logsumexp doc * update api doc * update api doc --- python/paddle/tensor/math.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py index 36d61fa0854..df0116c4c29 100755 --- a/python/paddle/tensor/math.py +++ b/python/paddle/tensor/math.py @@ -1078,7 +1078,8 @@ def logsumexp(x, axis=None, keepdim=False, name=None): logsumexp(x) = \\log\\sum exp(x) Args: - x (Tensor): The input Tensor with data type float32, float64. + x (Tensor): The input Tensor with data type float32 or float64, which + have no more than 4 dimensions. axis (int|list|tuple, optional): The axis along which to perform logsumexp calculations. ``axis`` should be int, list(int) or tuple(int). If ``axis`` is a list/tuple of dimension(s), logsumexp -- GitLab