jax.experimental.sparse.bcoo_fromdense

jax.experimental.sparse.bcoo_fromdense(mat, *, nse=None, n_batch=0, n_dense=0, index_dtype=<class 'jax.numpy.int32'>)

从稠密矩阵创建 BCOO 格式的稀疏矩阵。

参数：:

• mat (Array) – 要转换为 BCOO 的数组。

• nse (int | None | None) – 每个批次中指定的元素数量

• n_batch (int) – 批次维度的数量（默认值：0）

• n_dense (int) – 块维度的数量（默认值：0）

• index_dtype (DTypeLike) – 稀疏索引的 dtype（默认值：int32）

返回值：:

矩阵的 BCOO 表示。

返回类型：:

mat_bcoo