jax.experimental.sparse.bcoo_multiply_dense

jax.experimental.sparse.bcoo_multiply_dense#

jax.experimental.sparse.bcoo_multiply_dense(sp_mat, v)[source]#

An element-wise multiplication between a sparse and a dense array.

Parameters:
  • lhs – A BCOO-format array.

  • rhs – An ndarray.

  • sp_mat (BCOO)

  • v (Array)

Return type:

Array

Returns:

An ndarray containing the result.