jax.experimental.sparse.coo_todense

jax.experimental.sparse.coo_todense(mat)

将 COO 格式的稀疏矩阵转换为稠密矩阵。

参数:

mat (COO) – COO 矩阵

返回:

mat 的稠密版本

返回类型:

mat_dense