jax.experimental.sparse.eye

目录

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) – data-type, 可选。返回数组的类型。

  • index_dtype (DTypeLike) – (可选) 索引数组的 dtype。

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

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

  • sparse_format (str)

返回:

二维稀疏矩阵，第 k 个对角线上为 1。

返回类型:

I

目录