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