jax.experimental.sparse.bcoo_todense

jax.experimental.sparse.bcoo_todense(mat)

将批处理的稀疏矩阵转换为密集矩阵。

参数:

mat (BCOO) – BCOO 矩阵。

返回:

mat 的密集版本。

返回类型:

mat_dense