jax.numpy.unique_values

jax.numpy.unique_values(x, /, *, size=None, fill_value=None)

從 x 傳回唯一值。

numpy.unique_values() 的 JAX 實作；這等同於呼叫 jax.numpy.unique() 並將 equal_nan 設定為 True。

由於 unique_values 的輸出大小取決於資料，因此此函數通常與 jit() 和其他 JAX 轉換不相容。JAX 版本新增了選用的 size 引數，必須靜態指定此引數，才能在這些情況下使用 jnp.unique。

參數：

• x (ArrayLike) – 將從中提取唯一值的 N 維陣列。

• size (int | None | None) – 如果指定，則僅傳回前 size 個排序後的唯一元素。如果唯一元素少於 size 指示的數量，則傳回值將以 fill_value 填充。

• fill_value (ArrayLike | None | None) – 當指定 size 且元素數量少於指示的數量時，將剩餘條目填充 fill_value。預設為最小唯一值。

傳回：

形狀為 (n_unique,) 的陣列 values，其中包含來自 x 的唯一值。

傳回型別：

Array

另請參閱

• jax.numpy.unique()：用於計算唯一值的一般函數。

• jax.numpy.unique_values()：僅計算 values。

• jax.numpy.unique_counts()：僅計算 values 和 counts。

• jax.numpy.unique_inverse()：僅計算 values 和 inverse。

範例

在此範例中，我們計算一維陣列中的唯一值

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

有關 size 和 fill_value 引數的範例，請參閱 jax.numpy.unique()。