torch.ormqr

torch.ormqr(input, tau, other, left=True, transpose=False, *, out=None) → 张量

计算Householder矩阵的乘积与一般矩阵的矩阵-矩阵乘法。

将一个 $m \times n$ 矩阵 C（由 other 给出）与矩阵 Q 相乘， 其中 Q 使用 Householder 反射器 (input, tau) 表示。 有关更多详细信息，请参阅 正交或酉矩阵的表示。

如果 left 是 True，则计算 op(Q) 乘以 C，否则结果是 C 乘以 op(Q)。 当 left 是 True 时，隐式矩阵 Q 的大小为 $m \times m$。 否则，它的大小为 $n \times n$。 如果 transpose 是 True，则 op 是共轭转置操作，否则它是一个无操作。

支持float、double、cfloat和cdouble数据类型的输入。 还支持批量输入，并且如果输入是批量的，输出也会以相同的维度进行批量处理。

另请参阅

torch.geqrf() 可以用于形成矩阵 Q 的Householder表示 (input, tau)，从QR分解中得到。

注意

此函数支持反向传播，但仅当 (input, tau) 不需要梯度 和/或 tau.size(-1) 非常小时，速度才会快。 ``

Parameters

• 输入 (张量) – 形状为 (*, mn, k) 的张量，其中 * 表示零个或多个批次维度，mn 等于 m 或 n，取决于 left。

• tau (张量) – 形状为 (*, min(mn, k)) 的张量，其中 * 表示零个或多个批次维度。

• 其他 (张量) – 形状为 (*, m, n) 的张量，其中 * 表示零个或多个批次维度。

• left (bool) – 控制乘法的顺序。

• 转置 (布尔值) – 控制矩阵 Q 是否进行共轭转置。

Keyword Arguments

输出 (张量, 可选) – 输出张量。如果为无则忽略。默认值：无。