torch.hspmm

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

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

参数:

• mat1（张量）- 要进行矩阵乘法的第一个稀疏矩阵

• mat2（张量）- 要进行矩阵乘法的第二个步进矩阵

关键字参数:

输出（张量，可选）- 输出张量。