torch.sspaddmm
-
torch.sspaddmm(input, mat1, mat2, *, beta=1, alpha=1, out=None) → Tensor
将稀疏张量 mat1 与密集张量 mat2 相乘,然后将稀疏张量 input 加到结果上。
注意:此函数与 torch.addmm() 等价,但 input 和 mat1 是稀疏的。
- 参数:
输入(张量)- 要相加的稀疏矩阵
mat1(张量)- 要进行矩阵乘法的稀疏矩阵
mat2(张量)- 要进行矩阵乘法的稠密矩阵
- 关键字参数:
-