enp.compat.norm

enp.compat.norm#

etils.enp.compat.norm(
x: etils.enp.array_types.typing.FloatArray,
axis: int | None = None,
keepdims: bool = False,
) etils.enp.array_types.typing.FloatArray[source]

Like np.linalg.norm but auto-support jnp, tnp, np.