ActivationFunction

class openff.nagl.nn.ActivationFunction(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)[source]

Bases: Enum

Activation function options

Methods

get

get_value

get_function

Attributes

Identity

Tanh

ReLU

LeakyReLU

ELU

Sigmoid

ELU = <class 'torch.nn.modules.activation.ELU'>
Identity = <class 'torch.nn.modules.linear.Identity'>
LeakyReLU = <class 'torch.nn.modules.activation.LeakyReLU'>
ReLU = <class 'torch.nn.modules.activation.ReLU'>
Sigmoid = <class 'torch.nn.modules.activation.Sigmoid'>
Tanh = <class 'torch.nn.modules.activation.Tanh'>
classmethod get(name: str) → ActivationFunction[source]
classmethod get_function(name: str) → Callable[[tensor], Tensor][source]
classmethod get_value(name: str) → Callable[[tensor], Tensor][source]