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