Rate this Page
★★★★★

torch.hspmm#

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

Performs a matrix multiplication of asparse COO matrixmat1 and a strided matrixmat2. Theresult is a (1 + 1)-dimensionalhybrid COO matrix.

Parameters

  • mat1 (Tensor) – the first sparse matrix to be matrix multiplied

  • mat2 (Tensor) – the second strided matrix to be matrix multiplied

Keyword Arguments

out (Tensor,optional) – the output tensor.