Public API:
nt.stax
Dropout()
nt.empirical
nt.predict
nt.batch
nt.monte_carlo_kernel_fn
Internal:
nt.experimental
Kernel
Colab Examples:
Papers:
Other Resources:
Dropout.
Based on jax.example_libraries.stax.Dropout.
jax.example_libraries.stax.Dropout
rate (float) – Specifies the keep rate, e.g. rate=1 is equivalent to keeping all neurons.
float
rate
rate=1
mode (str) – Either "train" or "test".
str
"train"
"test"
Tuple[InitFn, ApplyFn, LayerKernelFn]
Tuple
InitFn
ApplyFn
LayerKernelFn
(init_fn, apply_fn, kernel_fn).
(init_fn, apply_fn, kernel_fn)