torch.hspmm

torch.hspmm(mat1, mat2, *, out=None) → Tensor

執行 稀疏 COO 矩陣 mat1 和跨步矩陣 mat2 的矩陣乘法。 結果是 (1 + 1) 維 混合 COO 矩陣。

參數

• mat1 (Tensor) – 要進行矩陣相乘的第一個稀疏矩陣

• mat2 (Tensor) – 要進行矩陣相乘的第二個跨步矩陣

關鍵字參數

out (Tensor, 選用) – 輸出 Tensor。