mlx.core.identity

identity(n: int, dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array

Create a square identity matrix.

參數:

• n (int) -- The number of rows and columns in the output.

• dtype (Dtype, optional) -- 輸出陣列的資料型別。默認值為 float32。

• stream (Stream, optional) -- Stream or device. Defaults to None.

回傳值:

An identity matrix of size n x n.

回傳值型別:

array