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 (Number, optional) – mat 的乘數 ($\beta$)

• alpha (Number, optional) – $mat1 @ mat2$ 的乘數 ($\alpha$)