jax.experimental.sparse.bcsr_fromdense

jax.experimental.sparse.bcsr_fromdense#

jax.experimental.sparse.bcsr_fromdense(mat, *, nse=None, n_batch=0, n_dense=0, index_dtype=<class 'jax.numpy.int32'>)[源代码][源代码]#

从密集矩阵创建BCSR格式的稀疏矩阵。

参数:

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

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

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

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

  • index_dtype (DTypeLike) – 稀疏索引的数据类型（默认：int32）

返回:

矩阵的BCSR表示。

返回类型:

mat_bcsr

目录