mlx.core.triu

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

Zeros the array below the given diagonal.

參數:

• 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.

回傳:

Array zeroed below the given diagonal

回傳型別:

array