neural_tangents.stax.Gaussian

neural_tangents.stax.Gaussian(a=1, b=-1)[source]

Elementwise Gaussian function a * np.exp(b * x**2).

Return type

Tuple[InitFn, ApplyFn, LayerKernelFn]

Returns

(init_fn, apply_fn, kernel_fn).