Public API:
nt.stax
Sin()
nt.empirical
nt.predict
nt.batch
nt.monte_carlo_kernel_fn
Internal:
nt.experimental
Kernel
Colab Examples:
Papers:
Other Resources:
Affine transform of Sin nonlinearity, i.e. a sin(b*x + c).
Sin
a sin(b*x + c)
a (float) – output scale.
float
b (float) – input scale.
c (float) – input phase shift.
tuple[InitFn, ApplyFn, LayerKernelFn]
tuple
InitFn
ApplyFn
LayerKernelFn
(init_fn, apply_fn, kernel_fn).
(init_fn, apply_fn, kernel_fn)