快捷方式

torch.sspaddmm

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

將稀疏張量 mat1 與稠密張量 mat2 進行矩陣相乘,然後將稀疏張量 input 加到結果中。

注意:此函式等同於 torch.addmm(),但 inputmat1 是稀疏張量。

引數
  • input (Tensor) – 要相加的稀疏矩陣

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

  • mat2 (Tensor) – 要進行矩陣乘法的稠密矩陣

關鍵字引數
  • beta (Number, optional) – mat 的乘數 (β\beta)

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

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

文件

訪問 PyTorch 的完整開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源