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