評價此頁

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) – 輸出張量。