optax.global_norm

Contents

optax.global_norm#

optax.global_norm(updates: optax.PyTree) Array[source]#

Compute the global norm across a nested structure of tensors.

Warning

Deprecated in favor of optax.tree.norm().

Parameters:

updates โ€“ A nested structure of tensors.

Returns:

The global L2 norm of the updates.