Spec-Zone.ru › PyTorch 1

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/1.13/generated/torch.sspaddmm.html

Spec-Zone.ru

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