mlx.core.clip

clip(a: array, /, a_min: scalar | array | None, a_max: scalar | array | None, *, stream: None | Stream | Device = None) → array

将数组的值裁剪到给定的最小值和最大值之间。

如果 a_min 或 a_max 为 None，则忽略相应的边界。 a_min 和 a_max 中至少有一个不能为 None。输入 a 和边界值必须能够互相广播。

参数:

• a (数组) – 输入数组。

• a_min (标量或数组或None) – 要裁剪到的最小值。

• a_max (标量或数组或None) – 要裁剪到的最大值。

返回:

裁剪后的数组。

返回类型:

数组