torch.hspmm

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

Выполняет умножение матрицы разреженной матрицы COO mat1 на матрицу со страйдовой раскладкой mat2. Результат представляет собой гибридную матрицу COO размерности (1 + 1).

Параметры:

• mat1 (Tensor) – первая разреженная матрица для умножения
• mat2 (Tensor) – вторая матрица со страйдовой раскладкой для умножения

Именованные аргументы:

out (Tensor, optional) – выходной тензор.