jax.numpy.ptp

内容

jax.numpy.ptp#

jax.numpy.ptp(a, axis=None, out=None, keepdims=False)[source]#

返回给定轴上的峰峰值范围。

JAX 实现 numpy.ptp().

参数：:

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

  • **axis** (*Axis* | *None*) – 可选，int 或 int 序列，默认值=None。计算范围的轴。如果为 None，则在扁平化的数组上计算范围。

  • **keepdims** (*bool*) – bool，默认值=False。如果为 True，则保留大小为 1 的缩减轴。

  • **out** (*None* | *None*) – JAX 未使用。

返回：:

包含输入沿指定轴的元素范围的数组。

返回类型：:

Array

示例

默认情况下，jnp.ptp 计算所有轴上的范围。

>>> x = jnp.array([[1, 3, 5, 2],
...                [4, 6, 8, 1],
...                [7, 9, 3, 4]])
>>> jnp.ptp(x)
Array(8, dtype=int32)

如果 axis=1，则计算轴 1 上的范围。

>>> jnp.ptp(x, axis=1)
Array([4, 7, 6], dtype=int32)

为了保留输入的维度，您可以设置 keepdims=True。

>>> jnp.ptp(x, axis=1, keepdims=True)
Array([[4],
       [7],
       [6]], dtype=int32)
目录