mlx.core.array.diag

array.diag(self, k: int = 0, *, stream: Stream | Device | None = None) → array

擷取對角線或建構對角矩陣。