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

tf_agents.trajectories.time_step.restart

View source on GitHub

Returns a TimeStep with step_type set equal to StepType.FIRST.

tf_agents.trajectories.time_step.restart(
    observation, batch_size=None
)

Used in the notebooks

Used in the tutorials

Args:

  • observation: A NumPy array, tensor, or a nested dict, list or tuple of arrays or tensors.
  • batch_size: (Optional) A python or tensorflow integer scalar.

Returns:

A TimeStep.