mlx.core.triu

triu(x: array, k: int, *, stream: None | Stream | Device = None) → array

将数组给定对角线以下的元素置零。

参数:

• x (array) – 输入数组。

• k (int, optional) – 2D 数组的对角线。默认为 0。

• stream (Stream, optional) – 流或设备。默认为 None。

返回值:

给定对角线以下元素已置零的数组

返回类型:

array