torch.sparse.addmm

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

此函数在前向传播中与 torch.addmm() 的功能完全相同，但它支持对稀疏 COO 矩阵 mat1 进行反向传播。当 mat1 是 COO Tensor 时，它必须具有 sparse_dim = 2。当输入是 COO Tensor 时，此函数也支持对两个输入进行反向传播。

支持 CSR 和 COO 存储格式。

注意

此函数不支持计算相对于 CSR 矩阵的导数。

参数

• mat (Tensor) – 一个待相加的稠密矩阵

• mat1 (Tensor) – 一个待相乘的稀疏矩阵

• mat2 (Tensor) – 一个待相乘的稠密矩阵

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

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