设计
模型库
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<>