ertk.pytorch.utils.get_activation

ertk.pytorch.utils.get_activation(act: str, *args, **kwargs) → Module

Get torch Module for activation function.

Parameters:

act: str

String representing the activation function.

*args, **kwargs: optional

Arguments and keyword arguments to pass to activation function module __init__().

Returns:

torch.nn.Module

The activation function as a Module.