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 (数字, 可选) – mat 的乘数 ($\beta$)

• alpha (Number, optional) – 用于 $mat1 @ mat2$ 的乘数 ($\alpha$)

• out (Tensor, 可选) – 指定输出张量。