jax.nn.standardize#
- jax.nn.standardize(x, axis=-1, mean=None, variance=None, epsilon=1e-05, where=None)[source]#
Normalizes an array by subtracting
mean
and dividing by \(\sqrt{\mathrm{variance}}\).- Parameters:
x (
Union
[Array
,ndarray
,bool_
,number
,bool
,int
,float
,complex
]) –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: