快捷键

torch.sspaddmm

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

将稀疏张量 mat1 与密集张量 mat2 相乘,然后将稀疏张量 input 加到结果上。

注意:此函数与 torch.addmm() 等价,但 inputmat1 是稀疏的。

参数:
  • 输入(张量)- 要相加的稀疏矩阵

  • mat1(张量)- 要进行矩阵乘法的稀疏矩阵

  • mat2(张量)- 要进行矩阵乘法的稠密矩阵

关键字参数:
  • beta(数字,可选)- matβ\beta )的乘数

  • alpha(数字,可选)- mat1@mat2mat1 @ mat2α\alpha )的乘数

  • 输出(张量,可选)- 输出张量。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源