torch.hspmm

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

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

Параметры:

• mat1 (Тензор) – первая разреженная матрица, подлежащая матричному умножению
• mat2 (Тензор) – вторая строчечная матрица, подлежащая матричному умножению

Ключевые аргументы:

out (Тензор, необязательно) – выходной тензор.