View source on GitHub |
Creates the chosen arm features info spec from the arm observation spec.
tf_agents.policies.utils.create_chosen_arm_features_info_spec(
observation_spec: tf_agents.typing.types.NestedTensorSpec
) -> tf_agents.typing.types.NestedTensorSpec