jax.experimental.sparse.eye

jax.experimental.sparse.eye(N, M=None, k=0, dtype=None, index_dtype='int32', sparse_format='bcoo', **kwds)

创建 2D 稀疏单位矩阵。

参数:

• N (int) – int。输出中的行数。

• M (int | None | None) – int，可选。输出中的列数。如果为 None，则默认为 N。

• k (int) – int，可选。对角线的索引：0（默认值）表示主对角线，正值表示上对角线，负值表示下对角线。

• dtype (DTypeLike | None | None) – 数据类型，可选。返回数组的数据类型。

• index_dtype (DTypeLike) – （可选）索引数组的数据类型。

• format – 指定矩阵格式的字符串（例如 [‘bcoo’]）。

• **kwds – 传递给特定格式的 _empty 构造函数的额外关键字。

• sparse_format (str)

返回:

沿第 k 条对角线为 1 的二维稀疏矩阵。

返回类型:

I