diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py index 36d61fa08546bf37f5f7a789f42ffb2f9e974e0e..df0116c4c29c297870fbd096ab4d842482630682 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