mlx.core.array.diag

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

Extract a diagonal or construct a diagonal matrix.