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

• alpha (Number, 可选) – $mat1 @ mat2$ 的乘数 ($\alpha$)

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