ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.trajectories.termination

Returns a TimeStep with step_type set to StepType.LAST.

Used in the notebooks

Used in the tutorials

observation A NumPy array, tensor, or a nested dict, list or tuple of arrays or tensors.
reward A NumPy array, tensor, or a nested dict, list or tuple of arrays or tensors.
outer_dims (optional) If provided, it will be used to determine the batch dimensions. If not, the batch dimensions will be inferred by reward's shape.

A TimeStep.