tf_agents.trajectories.to_transition_spec

Create a transition spec from a trajectory spec.

trajectory_spec An instance of Trajectory representing trajectory specs.

A tuple (time_steps, policy_steps, next_time_steps) specs.