tf_agents.trajectories.restart

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

View aliases

Main aliases

tf_agents.trajectories.time_step.restart

tf_agents.trajectories.restart(
    observation: tf_agents.typing.types.NestedTensorOrArray,
    batch_size: Optional[types.Int] = None,
    reward_spec: Optional[types.NestedSpec] = None
) -> tf_agents.trajectories.TimeStep

Used in the notebooks

Used in the tutorials
Policies Environments Tutorial on Multi Armed Bandits in TF-Agents

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. If not provided, the environment will not be considered as a batched environment.
reward_spec	If provided, the reward in the returned TimeStep will be compatible with the provided reward_spec.

Returns
A TimeStep.