Sspaddmm

Sparse matrix multiply and add: t + beta * m1 @ m2

Abstract Signature:

Sspaddmm(input: Tensor, mat1: Tensor, mat2: Tensor, beta: float = 1.0, alpha: float = 1.0)

PyTorchJAX (Core)

PyTorch

API: torch.sspaddmm

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '{beta} * {input} + {alpha} * jax.experimental.sparse.spmatmul({mat1}, {mat2})'