jax.experimental.sparse.bcoo_multiply_dense

jax.experimental.sparse.bcoo_multiply_dense#

jax.experimental.sparse.bcoo_multiply_dense(sp_mat, v)[source]#

稀疏数组和密集数组之间的逐元素乘法。

参数:
  • lhs – BCOO 格式的数组。

  • rhs – 一个 ndarray。

  • sp_mat (BCOO)

  • v (Array)

返回值:

包含结果的 ndarray。

返回类型:

Array