Public API:
nt.stax
ABRelu()
nt.empirical
nt.predict
nt.batch
nt.monte_carlo_kernel_fn
Internal:
nt.experimental
Kernel
Colab Examples:
Papers:
Other Resources:
ABReLU nonlinearity, i.e. a * min(x, 0) + b * max(x, 0).
a * min(x, 0) + b * max(x, 0)
a (float) – slope for x < 0.
float
x < 0
b (float) – slope for x > 0.
x > 0
do_stabilize (bool) – set to True for very deep networks.
bool
True
tuple[InitFn, ApplyFn, LayerKernelFn]
tuple
InitFn
ApplyFn
LayerKernelFn
(init_fn, apply_fn, kernel_fn).
(init_fn, apply_fn, kernel_fn)