mlx.core.tril

Contents

mlx.core.tril#

tril(x: array, k: int, *, stream: 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

Contents