neural_tangents.stax.Abs

neural_tangents.stax.Abs(do_stabilize=False)

Absolute value nonlinearity.

Parameters

do_stabilize (bool) – set to True for very deep networks.

Return type

Tuple[InitFn, ApplyFn, LayerKernelFn]

Returns

(init_fn, apply_fn, kernel_fn).