jax.experimental.sparse.bcoo_multiply_sparse#

jax.experimental.sparse.bcoo_multiply_sparse(lhs, rhs)[源代码]#

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

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

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

返回

一个包含结果的 BCOO 格式的数组。

返回类型

BCOO