From a7ef724dd59f752af02cb8ac66f620df9fc5e6e6 Mon Sep 17 00:00:00 2001 From: GaoWei8 <53294385+GaoWei8@users.noreply.github.com> Date: Fri, 27 Nov 2020 12:31:27 +0800 Subject: [PATCH] polish softamx doc (#29153) --- python/paddle/fluid/layers/nn.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index c07ea09064c..804e4e6d466 100755 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -1283,8 +1283,7 @@ def softmax(input, use_cudnn=True, name=None, axis=-1): Args: input (Tensor): The input tensor. A multi-dimension ``Tensor`` with type float32 or float64. use_cudnn (bool, optional): Use cudnn kernel or not, it is valid only when the cudnn \ - library is installed. To improve numerical stability, set use_cudnn to \ - False by default. + library is installed. To improve performance, set use_cudnn to True by default. name (str, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name` . Default: None. will be named automatically. Default: None. axis (int, optional): The index of dimension to perform softmax calculations, it should -- GitLab