mlx.core.eye

目录

mlx.core.eye#

eye(n: int, m: int | None = None, k: int = 0, dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#

创建一个单位矩阵或通用对角矩阵。

参数:

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

  • m (int, 可选) – 输出中的列数。默认为 n。

  • k (int, 可选) – 对角线的索引。默认为 0（主对角线）。

  • dtype (Dtype, 可选) – 输出数组的数据类型。默认为 float32。

  • stream (Stream, 可选) – 流或设备。默认为 None。

返回值:

一个数组，其中除第 k 条对角线上的值为一外，所有元素都为零。

返回值类型:

array

目录