tf_agents.specs.tensor_spec.add_outer_dim

Adds an outer dimension to the shape of input specs.

tf_agents.specs.tensor_spec.add_outer_dim(
    specs, dim=None
)

Used in the tutorials
Train a Deep Q Network with TF-Agents REINFORCE agent

Args
`specs`	Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
`dim`	Int, representing the outer dimension to be added to the TensorSpecs in specs.

Returns
Nested TensorSpecs with outer dimensions added to the shape of input specs.

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.

Args