评价此页

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) – 输出张量。