Ctrl+K

CELU

Contents

CELU#

class brainpy.dnn.CELU(alpha=1.0, inplace=False)[source]#

Applies the element-wise function:

\[\text{CELU}(x) = \max(0,x) + \min(0, \alpha * (\exp(x/\alpha) - 1))\]

More details can be found in the paper Continuously Differentiable Exponential Linear Units .

Parameters:

  • alpha (float) – the \(\alpha\) value for the CELU formulation. Default: 1.0

  • inplace (bool) – can optionally do the operation in-place. Default: False

Shape:

  • Input: \((*)\), where \(*\) means any number of dimensions.

  • Output: \((*)\), same shape as the input.

Examples:

>>> import brainpy as bp
>>> import brainpy.math as bm
>>> m = bp.dnn.CELU()
>>> input = bm.random.randn(2)
>>> output = m(input)
update(input)[source]#

The function to specify the updating rule.

Return type:

TypeVar(ArrayType, Array, Variable, TrainVar, Array, ndarray)

Contents