mindspore.nn.get_activation

mindspore.nn.get_activation(name, prim_name=None)

获取激活函数。

参数：

• name (str) - 激活函数名称。

• prim_name (Union[str, None]) - 算子名称，默认值： None 。

返回：

激活函数。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.nn as nn
>>> sigmoid = nn.get_activation('sigmoid')
>>> print(sigmoid)
Sigmoid<>