mlx.core.tril

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

将给定对角线上方的数组元素归零。

参数:

• x (array) – 输入数组。

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

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

返回值:

在给定对角线上方归零的数组

返回类型:

array