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