Compute the matrix or vector norm of an array.
The behavior depends on the axis argument, matching jax.numpy.linalg.norm():
axis
jax.numpy.linalg.norm()
ord
Input array.
Order of the norm (default: 2-norm for vectors, Frobenius norm for matrices).
The norm of x, reduced over the given axes.
x
Compute the matrix or vector norm of an array.
The behavior depends on the
axisargument, matchingjax.numpy.linalg.norm():axisis null andordis null, computes the 2-norm of the flattened array (any rank is allowed).axisis null andordis given, the input must be 1D (vector norm) or 2D (matrix norm).axisis an integer, computes a vector norm over that axis.axisis a pair of integers, computes a matrix norm over those axes.