Distribution API#

Base Distribution class#

Distribution

The base class for distribution over a random variable.

from_logits

Creates a Distribution from logits.

sample

Draw a sample from the distribution.

rsample

Draw a re-parameterized sample from the action distribution.

logp

The log-likelihood of the distribution computed at value

kl

The KL-divergence between two distributions.