torch.sparse.addmm

torch.sparse.addmm(mat, mat1, mat2, *, beta=1., alpha=1.) → Tensor

此函数在前向传播中与torch.addmm()执行相同的操作，但除了它支持稀疏COO矩阵mat1的反向传播。当mat1是COO张量时，必须满足sparse_dim = 2的要求。如果输入为COO张量，则此函数还支持对两个输入进行反向传播。

支持_CSR_和_COO_存储格式。

注意

此函数不支持计算与CSR矩阵相关的导数。

参数

• mat (张量) – 需要添加的密集矩阵

• mat1 (Tensor) – 待与之相乘的稀疏矩阵

• mat2 (Tensor) – 需要与另一个矩阵相乘的密集矩阵

• beta (数字, 可选) – mat 的乘数 ($\beta$)

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