pantea.models.nn package#

Submodules#

pantea.models.nn.activation module#

pantea.models.nn.activation.cos(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.exp(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.gaussian(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.harmonic(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.identity(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.logistic(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.relu(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.revlogistic(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.softplus(x)[source]#

Return type: ArrayImpl

pantea.models.nn.activation.tanh(x)[source]#

Return type: ArrayImpl

pantea.models.nn.initializer module#

class pantea.models.nn.initializer.UniformInitializer(weights_range)[source]#

Bases: object

Custom uniform initializer for the FLAX model.

pantea.models.nn.model module#

Module contents#