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
) -> 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.
|
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.
|
View source on GitHub