设计
规格
API
API映射
迁移指南
FAQ
RELEASE NOTES
获取激活函数。
name (str) - 激活函数名称。
prim_name (Union[str, None]) - 算子名称,默认:None。
激活函数。
Ascend GPU CPU
Ascend
GPU
CPU
样例:
>>> sigmoid = nn.get_activation('sigmoid') >>> print(sigmoid) Sigmoid<>