Module: tf_agents.specs.distribution_spec

View source on GitHub

Spec definition for tensorflow_probability.Distribution.

Classes

class DistributionSpec: Describes a tfp.distribution.Distribution.

Functions

deterministic_distribution_from_spec(...): Creates a Deterministic distribution_spec from a tensor_spec.

nested_distributions_from_specs(...): Builds a nest of distributions from a nest of specs.