tf_agents.trajectories.trajectory.to_transition_spec

View source on GitHub

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.