jax.experimental.sparse.csr_fromdense

jax.experimental.sparse.csr_fromdense#

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

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

参数:

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

  • nse (int | None) – mat 中指定条目的数量。如果未指定，将从输入矩阵中计算。

  • index_dtype (DTypeLike) – 稀疏索引的 dtype

返回:

矩阵的CSR表示。

返回类型:

mat_coo

目录