mindspore.nn.get_activation

mindspore.nn.get_activation(name, prim_name=None)[源代码]

获取激活函数。

参数:

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

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

返回:

激活函数。

支持平台:

Ascend GPU CPU

样例:

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