jax.numpy.unique_counts#

jax.numpy.unique_counts(x, /, *, size=None, fill_value=None)[原始碼]#

從 x 傳回唯一值，以及計數。

JAX 實作的 numpy.unique_counts()；這相當於呼叫 jax.numpy.unique() 並將 return_counts 和 equal_nan 設定為 True。

由於 unique_counts 的輸出大小取決於資料，因此此函式通常與 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 填充剩餘的條目。預設為最小唯一值。

傳回:

  • values:

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

  • counts:

    形狀為 (n_unique,) 的陣列。包含 x 中每個唯一值的出現次數。

傳回型別:

元組 (values, counts)，具有以下屬性

另請參閱

範例

在這裡，我們計算一維陣列中的唯一值

>>> x = jnp.array([3, 4, 1, 3, 1])
>>> result = jnp.unique_counts(x)

結果是一個 NamedTuple，具有兩個具名屬性。values 屬性包含陣列中的唯一值

>>> result.values
Array([1, 3, 4], dtype=int32)

counts 屬性包含輸入中每個唯一值的計數

>>> result.counts
Array([2, 2, 1], dtype=int32)

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