jax.experimental.sparse.eye

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

创建一个二维稀疏单位矩阵。

参数:

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

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

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

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

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

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

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

• sparse_format (str)

返回:

具有第 k 个对角线上值为 1 的二维稀疏矩阵。

返回类型:

I