mlx.core.tril

mlx.core.tril(x: array, k: int, *, stream: Union[None, Stream, Device] = None) → array

Zeros the array above the given diagonal.

Parameters:

• x (array) – input array.

• k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

• stream (Stream, optional) – Stream or device. Defaults to None.

Returns:

Array zeroed above the given diagonal

Return type:

array