From 84f9bc9221e5b1216d432d09e02dc088a043f5d9 Mon Sep 17 00:00:00 2001 From: iLeGend <824040212@qq.com> Date: Tue, 19 Dec 2023 18:29:50 +0800 Subject: [PATCH] add parameter value for thresholded_relu --- docs/api/paddle/nn/ThresholdedReLU_cn.rst | 5 +++-- docs/api/paddle/nn/functional/thresholded_relu_cn.rst | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/docs/api/paddle/nn/ThresholdedReLU_cn.rst b/docs/api/paddle/nn/ThresholdedReLU_cn.rst index 1b30962eaa1..45142859ffa 100644 --- a/docs/api/paddle/nn/ThresholdedReLU_cn.rst +++ b/docs/api/paddle/nn/ThresholdedReLU_cn.rst @@ -2,7 +2,7 @@ ThresholdedReLU ------------------------------- -.. py:class:: paddle.nn.ThresholdedReLU(threshold=1.0, name=None) +.. py:class:: paddle.nn.ThresholdedReLU(threshold=1.0, value=0.0, name=None) Thresholded ReLU 激活层 @@ -10,7 +10,7 @@ Thresholded ReLU 激活层 ThresholdedReLU(x) = \begin{cases} x, \text{if } x > threshold \\ - 0, \text{otherwise} + value, \text{otherwise} \end{cases} 其中,:math:`x` 为输入的 Tensor。 @@ -18,6 +18,7 @@ Thresholded ReLU 激活层 参数 :::::::::: - **threshold** (float,可选) - ThresholdedReLU 激活计算公式中的 threshold 值。默认值为 1.0。 + - **value** (float,可选) - ThresholdedReLU 激活计算公式中的 value 值。默认值为 0.0。 - **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。 形状 diff --git a/docs/api/paddle/nn/functional/thresholded_relu_cn.rst b/docs/api/paddle/nn/functional/thresholded_relu_cn.rst index 197aa530a97..e9662abdcab 100644 --- a/docs/api/paddle/nn/functional/thresholded_relu_cn.rst +++ b/docs/api/paddle/nn/functional/thresholded_relu_cn.rst @@ -3,7 +3,7 @@ thresholded_relu ------------------------------- -.. py:function:: paddle.nn.functional.thresholded_relu(x, threshold=1.0, name=None) +.. py:function:: paddle.nn.functional.thresholded_relu(x, threshold=1.0, value=0.0, name=None) thresholded relu 激活层。计算公式如下: @@ -11,7 +11,7 @@ thresholded relu 激活层。计算公式如下: thresholded\_relu(x) = \begin{cases} x, \text{if } x > threshold \\ - 0, \text{otherwise} + value, \text{otherwise} \end{cases} 其中,:math:`x` 为输入的 Tensor。 @@ -21,6 +21,7 @@ thresholded relu 激活层。计算公式如下: :::::::::: - **x** (Tensor) - 输入的 ``Tensor``,数据类型为:float32、float64。 - **threshold** (float,可选) - thresholded_relu 激活计算公式中的 threshold 值。默认值为 1.0。 + - **value** (float,可选) - thresholded_relu 激活计算公式中的 value 值。默认值为 0.0。 - **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。 返回