torch.sspaddmm

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

Matrix multiplies a sparse tensor mat1 with a dense tensor mat2, then adds the sparse tensor input to the result.

Note: This function is equivalent to torch.addmm(), except input and mat1 are sparse.

Parameters

• input (Tensor) – a sparse matrix to be added
• mat1 (Tensor) – a sparse matrix to be matrix multiplied
• mat2 (Tensor) – a dense matrix to be matrix multiplied

Keyword Arguments

• beta (Number, optional) – multiplier for mat ( $\beta$)
• alpha (Number, optional) – multiplier for $mat1 @ mat2$ ( $\alpha$)
• out (Tensor, optional) – the output tensor.