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 矩阵的导数。