tf_agents.agents.ppo.ppo_utils.get_distribution_params

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.

Args
`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.

Returns
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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.

tf_agents.agents.ppo.ppo_utils.get_distribution_params

Args

Returns