Spmm

Sparse matrix multiplication.

Abstract Signature:

Spmm(mat1: Tensor, mat2: Tensor)

PyTorch

API: torch.spmm
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.experimental.sparse.spmatmul
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.sparse.sparse_dense_matmul
Strategy: Direct Mapping
Official Docs ↗