View source on GitHub |
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 .
|