torch.ormqr

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

計算 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 dtypes 的輸入。 也支援批次輸入，如果輸入是批次的，則輸出也是具有相同維度的批次的。

另請參閱

torch.geqrf() 可用於從 QR 分解中形成矩陣 Q 的 Householder 表示 (input, tau)。

注意

此函數支援 backward，但僅在 (input, tau) 不需要梯度和/或 tau.size(-1) 非常小時才快速。

參數

• input (Tensor) – 形狀為 (*, mn, k) 的張量，其中 * 是零個或多個批次維度，mn 等於 m 或 n，具體取決於 left。

• tau (Tensor) – 形狀為 (*, min(mn, k)) 的張量，其中 * 是零個或多個批次維度。

• other (Tensor) – 形狀為 (*, m, n) 的張量，其中 * 是零個或多個批次維度。

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

• transpose (bool) – 控制矩陣 Q 是否經過共軛轉置。

關鍵字參數

out (Tensor, optional) – 輸出張量。 如果為 None 則忽略。 預設值：None。