cogwheel.utils.exp_normalize

cogwheel.utils.exp_normalize(lnprob, axis=-1)

Return normalized probabilities from unnormalized log probabilities.

Parameters:
lnprobfloat array

Natural log of the unnormalized probability.

axisint

Axis of lnprob along which probabilities sum to 1.