jax.experimental.sparse.bcoo_multiply_sparse

jax.experimental.sparse.bcoo_multiply_sparse(lhs, rhs)

兩個稀疏陣列的逐元素乘法。

參數:

• lhs (BCOO) – 一個 BCOO 格式的陣列。

• rhs (BCOO) – 一個 BCOO 格式的陣列。

返回:

一個包含結果的 BCOO 格式陣列。

返回類型:

BCOO