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$)