jax.experimental.sparse.coo_fromdense

jax.experimental.sparse.coo_fromdense#

jax.experimental.sparse.coo_fromdense(mat, *, nse=None, index_dtype=<class 'jax.numpy.int32'>)[source]#

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

参数：

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

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

  • index_dtype (DTypeLike) – 稀疏索引的数据类型

返回值：

矩阵的 COO 表示。

返回类型：

mat_coo

内容