- jax.numpy.bincount(x, weights=None, minlength=0, *, length=None)[source]#
Count number of occurrences of each value in array of non-negative ints.
LAX-backend implementation of
Jax adds the optional length parameter which specifies the output length, and defaults to
x.max() + 1. It must be specified for bincount to be compiled with non-static operands. Values larger than the specified length will be discarded. If length is specified, minlength will be ignored.
np.bincountraises an error if the input array contains negative values,
jax.numpy.bincountclips negative values to zero.
Original docstring below.
The number of bins (of size 1) is one larger than the largest value in x. If minlength is specified, there will be at least this number of bins in the output array (though it will be longer if necessary, depending on the contents of x). Each bin gives the number of occurrences of its index value in x. If weights is specified the input array is weighted by it, i.e. if a value
nis found at position
out[n] += weight[i]instead of
out[n] += 1.
out – The result of binning the input array. The length of out is equal to
- Return type:
ndarray of ints