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

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