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.