A class that stores useful data for performing data conversions.
tf_agents.agents.data_converter.DataContext(
time_step_spec: tf_agents.trajectories.time_step.TimeStep
,
action_spec: tf_agents.typing.types.NestedTensorSpec
,
info_spec: tf_agents.typing.types.NestedTensorSpec
)
Args |
time_step_spec
|
A nest of tf.TimeStep representing the time_steps.
|
action_spec
|
A nest of tf.TypeSpec representing the actions.
|
info_spec
|
A nest of tf.TypeSpec representing the policy's info.
(Typically this is the info emitted by the collect policy).
|
Raises |
TypeError
|
If any of the specs are not nests containing tf.TypeSpec
objects.
|
Attributes |
action_spec
|
|
info_spec
|
|
time_step_spec
|
|
trajectory_spec
|
|
transition_spec
|
|