mlx.core.array.diag

数组。对角线(自我, k： int = 0, *, 流： Optional[Union[Stream, Device]] = 无) → 数组

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