dask.array.eye

dask.array.eye

dask.array.eye(N, chunks='auto', M=None, k=0, dtype=<class 'float'>)[源代码]

返回一个二维数组,其对角线上为1,其他位置为0。

参数
N整数

输出中的行数。

int, str

如何对数组进行分块。必须是以下形式之一:

  • 一个像 1000 这样的块大小。

  • 字节大小,如“100 MiB”,这将选择一个均匀的块状形状

  • 单词 “auto” 的作用类似于上述,但它使用配置值 array.chunk-size 作为块大小

Mint, 可选

输出中的列数。如果为 None,则默认为 N

kint, 可选

对角线索引:0(默认值)指主对角线,正值指上对角线,负值指下对角线。

dtype数据类型,可选

返回数组的`数据类型`。

返回
形状为 (N,M) 的数组

一个数组,其中所有元素都等于零,除了第 k 条对角线,其值等于一。