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 tensors 時，此函數也支援兩個輸入的反向傳播。

支援 CSR 和 COO 儲存格式。

注意

此函數不支援對 CSR 矩陣計算導數。

參數

• mat (Tensor) – 要被加上的稠密矩陣

• mat1 (Tensor) – 要被乘上的稀疏矩陣

• mat2 (Tensor) – 要被乘上的稠密矩陣

• beta (Number, optional) – mat 的乘數 ($\beta$)

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