matrix_norm

Computes the matrix norm of a matrix (or a stack of matrices) x.

Abstract Signature:

matrix_norm(x: array, keepdims: bool, ord: Optional[Union[int, float, Literal[inf, Any, fro, nuc]]])

PyTorch

API: torch.linalg.matrix_norm
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.linalg.matrix_norm
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.linalg.matrix_norm
Strategy: Direct Mapping
Official Docs ↗