neural_tangents.stax.Erf

neural_tangents.stax.Erf(a=1.0, b=1.0, c=0.0)

Affine transform of Erf nonlinearity, i.e. a * Erf(b * x) + c.

Parameters:

• a (float) – output scale.

• b (float) – input scale.

• c (float) – output shift.

Return type:

tuple[InitFn, ApplyFn, LayerKernelFn]

Returns:

(init_fn, apply_fn, kernel_fn).