vector_normΒΆ

Computes the vector norm of a vector (or batch of vectors) x.

Abstract Signature:

vector_norm(x: array, axis: Optional[Union[int, Tuple[int, Ellipsis]]], keepdims: bool, ord: Union[int, float, Literal[inf, Any]])

PyTorch

API: torch.linalg.vector_norm
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.linalg.vector_norm
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.linalg.vector_norm
Strategy: Direct Mapping