Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.trajectories.trajectory.to_transition_spec

View source on GitHub

Create a transition spec from a trajectory spec.

tf_agents.trajectories.trajectory.to_transition_spec(
    trajectory_spec
)

Args:

  • trajectory_spec: An instance of Trajectory representing trajectory specs.

Returns:

A tuple (time_steps, policy_steps, next_time_steps) specs.