快捷方式

torch.sspaddmm

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

将稀疏张量 mat1 与密集张量 mat2 进行矩阵乘法,然后将稀疏张量 input 添加到结果中。

注意:此函数等效于 torch.addmm(),除了 inputmat1 是稀疏的。

参数
  • input (Tensor) – 要添加的稀疏矩阵

  • mat1 (Tensor) – 要进行矩阵乘法的稀疏矩阵

  • mat2 (Tensor) – 要进行矩阵乘法的密集矩阵

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

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

  • out (Tensor, 可选) – 输出张量。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源