torch.hspmm

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

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

Параметры

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

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

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