jax.experimental.sparse.bcoo_multiply_sparse

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.