jax.experimental.sparse.bcoo_multiply_dense#

jax.experimental.sparse.bcoo_multiply_dense(sp_mat, v)[原始碼]#

稀疏陣列和密集陣列之間的元素級乘法。

參數:

  • lhs – BCOO 格式陣列。

  • rhs – ndarray。

  • sp_mat (BCOO)

  • v (Array)

傳回:

包含結果的 ndarray。

傳回類型:

Array