tf_agents.trajectories.time_step.restart

View source on GitHub

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

Used in the notebooks

Used in the tutorials

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.

A TimeStep.