Attend the Women in ML Symposium on December 7 Register now

tf_agents.agents.ppo.ppo_utils.get_distribution_params

Stay organized with collections Save and categorize content based on your preferences.

Get the params for an optionally nested action distribution.

tf_agents.agents.ppo.ppo_utils.get_distribution_params(
    nested_distribution: tf_agents.typing.types.NestedDistribution,
    legacy_distribution_network: bool
) -> tf_agents.typing.types.NestedTensor

Only returns parameters that have tf.Tensor values.

nested_distribution The nest of distributions whose parameter tensors to extract.
legacy_distribution_network Whether this distribution was generated by a (legacy) DistributionNetwork. If False, expect the distribution was generated by a regular Network object that emits distributions.

A nest of distribution parameters. Each leaf is a dict corresponding to one distribution, with keys as parameter name and values as tensors containing parameter values.