forked from mindspore-Ecosystem/mindspore
15 lines
329 B
ReStructuredText
15 lines
329 B
ReStructuredText
|
mindspore.nn.get_activation
|
|||
|
|
============================
|
|||
|
|
|
|||
|
|
.. py:function:: mindspore.nn.get_activation(name, prim_name=None)
|
|||
|
|
|
|||
|
|
获取激活函数。
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
- **name** (str) - 激活函数名称。
|
|||
|
|
- **prim_name** (Union[str, None]) - 算子名称，默认：None。
|
|||
|
|
|
|||
|
|
**返回：**
|
|||
|
|
|
|||
|
|
激活函数。
|