sciai.architecture.get_activation

View Source On Gitee
sciai.architecture.get_activation(activation)[source]

Get the activation function according to its name.

Parameters

activation (str) – The name of the activation function.

Returns

Callable, The corresponding activation function.

Supported Platforms:

Ascend GPU CPU

Examples

>>> from sciai.architecture.activation import get_activation
>>> sigmoid = get_activation('sigmoid')
>>> print(sigmoid)
Sigmoid<>