Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix docs problems for nn.Threshold

This commit is contained in:
lvyufeng 2022-05-25 17:38:53 +08:00
parent f3148dbe6e
commit 708cad7bc3
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/api/api_python/nn/mindspore.nn.Threshold.rst
View File

@ -1,7 +1,7 @@
mindspore.nn.Threshold
=============================
.. py:class:: mindspore.nn.Threshold
.. py:class:: mindspore.nn.Threshold(threshold, value)
Threshold激活函数，按元素计算输出。

4
mindspore/python/mindspore/nn/layer/activation.py
View File

@ -1084,8 +1084,8 @@ class Threshold(Cell):
\end{cases}
Args:
threshold: The value to threshold at.
value: The value to replace with when element is less than threshold.
threshold (Union[int, float]): The value to threshold at.
value (Union[int, float]): The value to replace with when element is less than threshold.
Inputs:
- **input_x** (Tensor) - The input of Threshold with data type of float16 or float32.