jax.experimental.sparse.bcoo_multiply_sparse#

jax.experimental.sparse.bcoo_multiply_sparse(lhs, rhs)[source]#

An element-wise multiplication of two sparse arrays.

Parameters
  • lhs (BCOO) – A BCOO-format array.

  • rhs (BCOO) – A BCOO-format array.

Return type

BCOO

Returns

An BCOO-format array containing the result.