快捷方式

torch.ormqr

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

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

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

如果 leftTrue,则计算 op(Q) 乘以 C,否则结果为 C 乘以 op(Q)。当 leftTrue 时,隐式矩阵 Q 的大小为 m×mm \times m。否则,其大小为 n×nn \times n。如果 transposeTrue,则 op 为共轭转置运算,否则为无操作。

支持 float、double、cfloat 和 cdouble 数据类型。还支持批处理输入,并且如果输入是批处理的,则输出也是批处理的,具有相同的维度。

另请参阅

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

注意

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

参数
  • input (张量) – 形状为 (*, mn, k) 的张量,其中 * 为零个或多个批处理维度,mn 等于 mn,具体取决于 left

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

  • other (张量) – 形状为 (*, m, n) 的张量,其中 * 为零个或多个批处理维度。

  • left (布尔值) – 控制乘法的顺序。

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

关键字参数

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源