快捷方式

torch.sparse.addmm

torch.sparse.addmm(mat, mat1, mat2, *, beta=1., alpha=1.) Tensor

此函数在正向传播中与 torch.addmm() 执行完全相同的事情,只是它支持稀疏 COO 矩阵 mat1 的反向传播。当 mat1 是 COO 张量时,它必须具有 sparse_dim = 2。当输入是 COO 张量时,此函数还支持两个输入的反向传播。

支持 CSR 和 COO 两种存储格式。

注意

此函数不支持计算相对于 CSR 矩阵的导数。

参数
  • mat (Tensor) – 要添加的密集矩阵

  • mat1 (Tensor) – 要乘以的稀疏矩阵

  • mat2 (Tensor) – 要乘以的密集矩阵

  • beta (Number, optional) – mat 的乘数 (β\beta)

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

文档

访问 PyTorch 的全面的开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源