neural_tangents.stax.Exp

neural_tangents.stax.Exp(a=1, b=1)[source]

Elementwise natural exponent function a * jnp.exp(b * x).

Return type:

tuple[InitFn, ApplyFn, LayerKernelFn]

Returns:

(init_fn, apply_fn, kernel_fn).