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，则上三角形将填充下三角形的转置，并且整个矩阵有效。