`get_activation`

python

get_activation(activation_name: str) -> type[nn.Module]

Resolve an activation name to the corresponding PyTorch module type.

Parameters:

activation_name (str) – Activation name such as "relu" or "tanh".

Returns:

type[Module] – Activation module type.