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 (Tensor) – 要添加的稠密矩阵

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

• mat2 (Tensor) – 要相乘的稠密矩阵

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

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