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.