tf_agents.utils.nest_utils.unbatch_nested_tensors
Remove the batch dimension if needed from nested tensors using their specs.
tf_agents.utils.nest_utils.unbatch_nested_tensors(
tensors, specs=None
)
If specs is None, the first dimension of each tensor will be removed.
If specs are provided, each tensor is compared to the corresponding spec,
and the first dimension is removed only if the tensor was batched.
Args |
tensors
|
Nested list/tuple or dict of batched Tensors.
|
specs
|
Nested list/tuple or dict of TensorSpecs, describing the shape of the
non-batched Tensors.
|
Returns |
A nested non-batched version of each tensor.
|
Raises |
ValueError
|
if the tensors and specs have incompatible dimensions or shapes.
|
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.
[null,null,["Last updated 2024-04-26 UTC."],[],[]]