jax.numpy.linalg.matrix_norm

jax.numpy.linalg.matrix_norm(x, /, *, keepdims=False, ord='fro')

计算矩阵或矩阵堆栈的范数。

JAX 实现的 numpy.linalg.matrix_norm()

参数:

• x (ArrayLike) – 形状为 (..., M, N) 的数组，用于计算范数。

• keepdims (bool) – 如果为真，在输出中保留降维后的维度。

• ord (str) – 指定范数类型的字符串或整数；默认是Frobenius范数。有关可用选项的详细信息，请参见 numpy.linalg.norm()。

返回:

包含 x 范数的数组。如果 keepdims 为 False，形状为 x.shape[:-2]，如果 keepdims 为 True，形状为 (..., 1, 1)。

返回类型:

Array

参见

• jax.numpy.linalg.vector_norm(): 向量或向量堆栈的范数。

• jax.numpy.linalg.norm(): 更一般的矩阵或向量范数。

示例

>>> x = jnp.array([[1, 2, 3],
...                [4, 5, 6],
...                [7, 8, 9]])
>>> jnp.linalg.matrix_norm(x)
Array(16.881943, dtype=float32)