torch.sspaddmm

torch.sspaddmm(input, mat1, mat2, *, beta=1, alpha=1, out=None) → Tensor

將稀疏張量 mat1 與稠密張量 mat2 相乘，然後將稀疏張量 input 加到結果中。

注意：此函式等同於 torch.addmm()，除了 input 和 mat1 是稀疏矩陣。

參數

• input (Tensor) – 要加入的稀疏矩陣

• mat1 (Tensor) – 要進行矩陣相乘的稀疏矩陣

• mat2 (Tensor) – 要進行矩陣相乘的密集矩陣

關鍵字參數

• beta (Number, optional) – mat 的乘數 ($\beta$)

• alpha (Number, optional) – $mat1 @ mat2$ 的乘數 ($\alpha$)

• out (Tensor, optional) – 輸出的張量。