Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.agents.ppo.ppo_utils.get_distribution_params

View source on GitHub

Get the params for an optionally nested action distribution.

tf_agents.agents.ppo.ppo_utils.get_distribution_params(
    nested_distribution
)

Only returns parameters that have tf.Tensor values.

Args:

  • nested_distribution: The nest of distributions whose parameter tensors to extract.

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.