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

!34938 fix docs problems for nn.Threshold 

Merge pull request !34938 from 吕昱峰（Nate.River）/code_docs_threshold
This commit is contained in:
i-robot 2022-05-26 04:26:33 +00:00 committed by Gitee
parent a29344460c 708cad7bc3
commit e960f5beb7
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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.