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)[source]#

创建二维稀疏单位矩阵。

参数：

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

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

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

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

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

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

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

  • sparse_format (str)

返回:

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

返回类型:

I

内容