mlx.core.array.diag

array.diag(self, k: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array

提取对角线或构造对角矩阵。