jax.numpy.histogram_bin_edges

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

计算直方图的箱体边缘。

JAX 对 numpy.histogram_bin_edges() 的实现。

参数:

• a (ArrayLike) – 需要进行分箱的值的数组

• bins (ArrayLike) – 指定直方图中的箱子数量（默认：10）。

• range (None | Array | Sequence[ArrayLike] | None) – 标量的元组。指定数据的范围。如果未指定，则从数据推断范围。

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

返回:

直方图的箱子边缘数组。

返回类型:

Array

另请参阅

• jax.numpy.histogram()：计算一维直方图。

• jax.numpy.histogram2d()：计算二维直方图。

• 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)