mlx.core.tri

目录

mlx.core.tri#

tri(n: int, m: int, k: int, dtype: Dtype | None = None, *, stream: None | Stream | Device =None) → array#

创建一个二维数组，其对角线及其下方的元素为 1，其余元素为 0。

参数:

  • n (int) – 输出数组的行数。

  • m (int, 可选) – 输出数组的列数。默认为 None。

  • k (int, 可选) – 二维数组的对角线索引。默认为 0。

  • dtype (Dtype, 可选) – 输出数组的数据类型。默认为 float32。

  • stream (Stream, 可选) – 流或设备。默认为 None。

返回值:

一个二维数组，其下三角形区域填充 1，其余区域填充 0

返回类型:

array

目录