mlx.core.tri#
- tri(n: int, m: int, k: int, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) array#
An array with ones at and below the given diagonal and zeros elsewhere.
- 參數:
n (int) -- The number of rows in the output.
m (int, optional) -- The number of cols in the output. Defaults to
None.k (int, optional) -- The diagonal of the 2-D array. Defaults to
0.dtype (Dtype, optional) -- Data type of the output array. Defaults to
float32.stream (Stream, optional) -- Stream or device. Defaults to
None.
- 回傳:
Array with its lower triangle filled with ones and zeros elsewhere
- 回傳型別: