tf_agents.trajectories.to_transition_spec

Create a transition spec from a trajectory spec.

View aliases

Main aliases

tf_agents.trajectories.trajectory.to_transition_spec

tf_agents.trajectories.to_transition_spec(
    trajectory_spec: tf_agents.trajectories.Trajectory
) -> tf_agents.trajectories.Transition

Note: since trajectories do not include the policy step's state (except in special cases where the policy chooses to store this in the info field), the returned transition.action_spec.state field will be an empty tuple.

Args
trajectory_spec	An instance of Trajectory representing trajectory specs.

Returns
A tuple (time_steps, policy_steps, next_time_steps) specs.