tf_agents.specs.tensor_spec.remove_outer_dims_nest

Removes the specified number of outer dimensions from the input spec nest.

tf_agents.specs.tensor_spec.remove_outer_dims_nest(
    specs, num_outer_dims
)

Args
`specs`	Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
`num_outer_dims`	(int) Number of outer dimensions to remove.

Returns
Nested TensorSpecs with outer dimensions removed from the input specs.

Raises
Value error if a spec in the nest has shape rank less than `num_outer_dims`.

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