jax.numpy.clip#

jax.numpy.clip(arr=None, /, min=None, max=None, *, a=Deprecated, a_min=Deprecated, a_max=Deprecated)[原始碼]#

將陣列數值裁剪到指定範圍。

JAX 實作的 numpy.clip()。

參數:

  • arr (ArrayLike | None) – 要裁剪的 N 維陣列。

  • min (ArrayLike | None) – 裁剪範圍的可選最小值；如果 None (預設值)，則結果將不會裁剪到任何最小值。如果指定，它應該與 arr 和 max 廣播相容。

  • max (ArrayLike | None) – 裁剪範圍的可選最大值；如果 None (預設值)，則結果將不會裁剪到任何最大值。如果指定，它應該與 arr 和 min 廣播相容。

  • a (ArrayLike | DeprecatedArg) – 已棄用的 arr 參數別名。如果使用，將導致 DeprecationWarning。

  • a_min (ArrayLike | None | DeprecatedArg) – 已棄用的 min 參數別名。如果使用，將導致 DeprecationWarning。

  • a_max (ArrayLike | None | DeprecatedArg) – 已棄用的 max 參數別名。如果使用，將導致 DeprecationWarning。

返回:

一個包含 arr 值的陣列，其中小於 min 的值設定為 min，大於 max 的值設定為 max。

返回類型:

陣列

參見

範例

>>> arr = jnp.array([0, 1, 2, 3, 4, 5, 6, 7])
>>> jnp.clip(arr, 2, 5)
Array([2, 2, 2, 3, 4, 5, 5, 5], dtype=int32)