torch.ormqr¶
- torch.ormqr(input, tau, other, left=True, transpose=False, *, out=None) Tensor ¶
计算 Householder 矩阵乘积与一般矩阵的矩阵乘法。
将一个 矩阵 C(由
other
给定)与矩阵 Q 相乘,其中 Q 使用 Householder 反射器 (input, tau) 表示。有关更多详细信息,请参阅 正交或酉矩阵的表示。如果
left
为 True,则计算 op(Q) 乘以 C,否则结果为 C 乘以 op(Q)。当left
为 True 时,隐式矩阵 Q 的大小为 。否则,其大小为 。如果transpose
为 True,则 op 为共轭转置运算,否则为无操作。支持 float、double、cfloat 和 cdouble 数据类型。还支持批处理输入,并且如果输入是批处理的,则输出也是批处理的,具有相同的维度。
另请参阅
torch.geqrf()
可用于从 QR 分解中形成矩阵 Q 的 Householder 表示 (input, tau)。注意
此函数支持反向传播,但仅当
(input, tau)
不需要梯度和/或tau.size(-1)
非常小时才快速。``