mlx.core.triu

Contents

mlx.core.triu#

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

Zeros the array below 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 below the given diagonal

Return type:

array

Contents