jax.lax.linalg.symmetric_product

jax.lax.linalg.symmetric_product(a_matrix, c_matrix, *, alpha=1.0, beta=0.0, symmetrize_output=False)

对称积。

计算对称积

\[\alpha \, A \, A^T + \beta \, C\]

其中 \(A\) 是一个矩形矩阵，\(C\) 是一个对称矩阵。

参数:

• a_matrix (ArrayLike) – 形状为 [..., m, n] 的矩阵批次。

• c_matrix (ArrayLike) – 形状为 [..., m, m] 的矩阵批次。

• alpha (float) – 标量。

• beta (float) – 标量。

• symmetrize_output (bool) – 如果为 True，则输出的上三角将替换为其转置。

返回:

形状为 [..., m, m] 的矩阵批次，其中仅保证下三角在所有平台上都包含正确的值。如果 symmetrize_output 为 True，则上三角将填充下三角的转置，并且整个矩阵都有效。