Toggle navigation

.rst .pdf

Warning

This page was created from a pull request (#9655).

jax.nn.normalize

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

Normalizes an array by subtracting mean and dividing by sqrt(var).

Parameters

• x (Any) –

• axis (Union[int, Tuple[int, …], None]) –

• mean (Optional[Any]) –

• variance (Optional[Any]) –

• epsilon (Any) –

• where (Optional[Any]) –

Return type

Any

previous

jax.nn.logsumexp

next

jax.nn.one_hot