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.