evorl.distribution

Module Contents

Classes

TanhTransformedDistribution

Distribution followed by tanh. from acme.

Functions

get_categorical_dist	Get a categorical distribution.
get_tanh_norm_dist	Get a tanh transformed normal distribution.
get_trancated_norm_dist	Get a truncated normal distribution.

Data

tfd

API

class evorl.distribution.TanhTransformedDistribution(distribution, threshold=0.999, validate_args=False)

Bases: evorl.distribution.tfd

Distribution followed by tanh. from acme.

entropy(seed=None)

log_prob(event)

mode()

evorl.distribution.get_categorical_dist(logits: jax.Array)

Get a categorical distribution.

evorl.distribution.get_tanh_norm_dist(loc: jax.Array, scale: jax.Array, min_scale: float = 0.001)

Get a tanh transformed normal distribution.

evorl.distribution.get_trancated_norm_dist(loc, scale, low, high)

Get a truncated normal distribution.

evorl.distribution.tfd

None