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）- 要相乘的矩阵

• beta（数字，可选）- mat （ $\beta$ ）的乘数

• alpha（数字，可选）- $mat1 @ mat2$ （ $\alpha$ ）的乘数