tf_agents.trajectories.restart
Stay organized with collections
Save and categorize content based on your preferences.
Returns a TimeStep
with step_type
set equal to StepType.FIRST
.
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
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 .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-04-26 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-04-26 UTC."],[],[]]