Dsmm ==== Performs a matrix multiplication of the sparse matrix input and the dense matrix mat2. **Abstract Signature:** ``Dsmm(input: Tensor, mat2: Tensor)`` .. raw:: html

PyTorch

API: torch.dsmm
Strategy: Direct Mapping

JAX (Core)

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

TensorFlow

API: tf.sparse.sparse_dense_matmul
Strategy: Direct Mapping

Flax NNX

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

PaxML / Praxis

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