torch.smm

torch.smm(input, mat) → Tensor

Выполняет умножение разреженной матрицы input на плотную матрицу mat.

Параметры:

• input (Tensor) – разреженная матрица для умножения
• mat (Tensor) – плотная матрица для умножения