jax.experimental.sparse.bcoo_multiply_sparse

jax.experimental.sparse.bcoo_multiply_sparse#

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

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

参数:
  • lhs (BCOO) – 一个 BCOO 格式的数组。

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

返回:

包含结果的 BCOO 格式数组。

返回类型:

BCOO