jax.numpy.唯一值

jax.numpy.唯一值#

jax.numpy.unique_values(x, /, *, size=None, fill_value=None)[源代码][源代码]#

从 x 中返回唯一值，以及索引、逆索引和计数。

JAX 实现的 numpy.unique_values()；这等同于调用 jax.numpy.unique() 并将 equal_nan 设置为 True。

由于 unique_values 的输出大小取决于数据，该函数通常不兼容 jit() 和其他 JAX 转换。JAX 版本添加了可选的 size 参数，必须在 jnp.unique 用于此类上下文中时静态指定该参数。

参数:

  • x (ArrayLike) – 将从中提取唯一值的 N 维数组。

  • size (int | None) – 如果指定，则仅返回前 size 个排序后的唯一元素。如果唯一元素的数量少于 size 指示的数量，则返回值将用 fill_value 填充。

  • fill_value (ArrayLike | None) – 当指定了 size 且元素数量少于指定数量时，用 fill_value 填充剩余的条目。默认为最小唯一值。

返回:

形状为 (n_unique,) 的数组 values 包含 x 中的唯一值。

返回类型:

Array

参见

示例

这里我们计算一维数组中的唯一值：

>>> x = jnp.array([3, 4, 1, 3, 1])
>>> jnp.unique_values(x)
Array([1, 3, 4], dtype=int32)

关于 size 和 fill_value 参数的示例，请参见 jax.numpy.unique()。

目录