Spec-Zone.ru › PyTorch 2

torch.sspaddmm

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

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

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

Параметры
  • input (Тензор) – разреженная матрица, которая должна быть добавлена
  • mat1 (Тензор) – разреженная матрица, которая должна быть умножена
  • mat2 (Тензор) – плотная матрица, которая должна быть умножена
Ключевые аргументы
  • beta (Число, необязательно) – множитель для mat (β\beta)
  • alpha (Число, необязательно) – множитель для mat1@mat2mat1 @ mat2 (α\alpha)
  • out (Тензор, необязательно) – тензор вывода.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.sspaddmm.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API