torch.sspaddmm

torch.sspaddmm(input, mat1, mat2, *, beta=1, alpha=1, out=None) → Tensor

Умножает разреженный тензор mat1 на плотный тензор mat2, а затем прибавляет разреженный тензор input к результату.

Примечание: эта функция эквивалентна torch.addmm(), за исключением того, что input и mat1 являются разреженными.

Параметры:

• input (Tensor) – разреженная матрица, которую нужно прибавить
• mat1 (Tensor) – разреженная матрица, которую нужно умножить на матрицу
• mat2 (Tensor) – плотная матрица, которую нужно умножить на матрицу

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

• beta (Number, optional) – множитель для mat ($\beta$)
• alpha (Number, optional) – множитель для $mat1 @ mat2$ ($\alpha$)
• out (Tensor, optional) – выходной тензор.