Spmm ==== Sparse matrix multiplication. **Abstract Signature:** ``Spmm(mat1: Tensor, mat2: Tensor)`` .. raw:: html

PyTorch

API: torch.spmm
Strategy: Direct Mapping

JAX (Core)

API: jax.experimental.sparse.spmatmul
Strategy: Direct Mapping

TensorFlow

API: tf.sparse.sparse_dense_matmul
Strategy: Direct Mapping