asteroid.masknn.activations module

class asteroid.masknn.activations.Swish

Bases: sphinx.ext.autodoc.importer._MockObject

forward(x)

asteroid.masknn.activations.gelu()

asteroid.masknn.activations.get(identifier)

Returns an activation function from a string. Returns its input if it is callable (already an activation for example).

Parameters:identifier (str or Callable or None) – the activation identifier. Returns:nn.Module or None

asteroid.masknn.activations.leaky_relu()

asteroid.masknn.activations.linear()

asteroid.masknn.activations.prelu()

asteroid.masknn.activations.register_activation(custom_act)

Register a custom activation, gettable with activation.get.

Parameters:custom_act – Custom activation function to register.

asteroid.masknn.activations.relu()

asteroid.masknn.activations.sigmoid()

asteroid.masknn.activations.softmax(dim=None)

asteroid.masknn.activations.swish()

asteroid.masknn.activations.tanh()