jax.numpy.histogram_bin_edges

jax.numpy.histogram_bin_edges(a, bins=10, range=None, weights=None)

計算直方圖的 bin 邊緣。

JAX 版本的 numpy.histogram_bin_edges()。

參數:

• a (ArrayLike) – 要分箱的值的陣列

• bins (ArrayLike) – 指定直方圖中的 bin 數量 (預設值：10)。

• range (None | Array | Sequence[ArrayLike] | None) – 純量元組。指定資料的範圍。如果未指定，則從資料推斷範圍。

• weights (ArrayLike | None | None) – JAX 未使用。

傳回:

直方圖的 bin 邊緣陣列。

傳回型別:

Array

另請參閱

• jax.numpy.histogram()：計算 1D 直方圖。

• jax.numpy.histogram2d()：計算 2D 直方圖。

• jax.numpy.histogramdd()：計算 N 維直方圖。

範例

>>> a = jnp.array([2, 5, 3, 6, 4, 1])
>>> jnp.histogram_bin_edges(a, bins=5)
Array([1., 2., 3., 4., 5., 6.], dtype=float32)
>>> jnp.histogram_bin_edges(a, bins=5, range=(-10, 10))  
Array([-10.,  -6.,  -2.,   2.,   6.,  10.], dtype=float32)