jax.nn.standardize

jax.nn.standardize(x, axis=-1, mean=None, variance=None, epsilon=1e-05, where=None)

Normalizes an array by subtracting mean and dividing by \(\sqrt{\mathrm{variance}}\).

Parameters:

• x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

• axis (Union[int, tuple[int, ...], None]) –

• mean (Union[Array, ndarray, bool_, number, bool, int, float, complex, None]) –

• variance (Union[Array, ndarray, bool_, number, bool, int, float, complex, None]) –

• epsilon (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

• where (Union[Array, ndarray, bool_, number, bool, int, float, complex, None]) –

Return type:

Array