HistogramBinEdges

Function to calculate only the edges of the bins used by the histogram function.

Abstract Signature:

HistogramBinEdges(a: Array, bins: Union[int, str] = 10, range: Optional[Tuple[float, float]], weights: Optional[Array])

PyTorch

API: —
Strategy: Custom / Partial

JAX (Core)

API: jax.numpy.histogram_bin_edges
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.histogram_bin_edges
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.histogram_bin_edges
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.numpy.histogram_bin_edges
Strategy: Direct Mapping
Official Docs ↗