jactorch.functional.probability

Probability distributions related functions.

Functions

check_prob_normalization(p[, dim, atol])	Check if the probability is normalized along a specific dimension.
normalize_prob(a[, dim])	Perform 1-norm along the specific dimension.

Functions

check_prob_normalization(p, dim=-1, atol=1e-5)

Check if the probability is normalized along a specific dimension.

normalize_prob(a, dim=-1)

Perform 1-norm along the specific dimension.