neural_tangents.stax.Sign

neural_tangents.stax.Sign()[source]

Sign function.

Return type:

tuple[InitFn, ApplyFn, LayerKernelFn]

Returns:

(init_fn, apply_fn, kernel_fn).