Shortcuts

torch.sspaddmm

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

矩阵将稀疏张量 mat1 与密集张量 mat2 相乘,然后将稀疏张量 input 加到结果中。

注意:此函数等同于 torch.addmm(),除了 inputmat1 是稀疏的。

Parameters
  • 输入 (张量) – 要添加的稀疏矩阵

  • mat1 (张量) – 一个用于矩阵乘法的稀疏矩阵

  • mat2 (张量) – 需要进行矩阵乘法的稠密矩阵

Keyword Arguments
  • beta (数字, 可选) – mat 的乘数 (β\beta)

  • alpha (数字, 可选) – mat1@mat2mat1 @ mat2 的乘数 (α\alpha)

  • 输出 (张量, 可选) – 输出张量。

优云智算