jax.experimental.sparse.bcoo_multiply_sparse

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

两个稀疏数组的逐元素乘法。

参数:

• lhs (BCOO) – BCOO 格式的数组。

• rhs (BCOO) – BCOO 格式的数组。

返回:

包含结果的 BCOO 格式数组。

返回类型:

BCOO