torch.hspmm

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

对一个稀疏COO矩阵 mat1 和一个跨步矩阵 mat2 进行矩阵乘法。结果是一个(1 + 1)维的混合COO矩阵。

Parameters

• mat1 (张量) – 第一个要进行矩阵乘法的稀疏矩阵

• mat2 (张量) – 要进行矩阵乘法的第二个跨步矩阵

Keyword Arguments

输出 (张量, 可选) – 输出张量。