torch.hspmm

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

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

引數

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

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

關鍵字引數

out (Tensor, optional) – 輸出張量。