mlx.core.array.diag

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

Extract a diagonal or construct a diagonal matrix.