torch.sparse.addmm

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

此函数在前向传播中与torch.addmm()完全相同， 除了它支持稀疏COO矩阵mat1的反向传播。 当mat1是COO张量时，它必须具有sparse_dim = 2。 当输入是COO张量时，此函数还支持两个输入的反向传播。

支持CSR和COO存储格式。

注意

此函数不支持对CSR矩阵进行导数计算。

Parameters

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

• mat1 (张量) – 要相乘的稀疏矩阵

• mat2 (张量) – 要相乘的密集矩阵

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

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