tf_agents.agents.ppo.ppo_utils.get_distribution_params

Get the params for an optionally nested action distribution.

Only returns parameters that have tf.Tensor values.

nested_distribution The nest of distributions whose parameter tensors to extract.

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.