View source on GitHub |
Adds outer dimensions to the shape of input specs.
tf_agents.specs.tensor_spec.add_outer_dims_nest(
specs, outer_dims
)
Args | |
---|---|
specs
|
Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors. |
outer_dims
|
a list or tuple, representing the outer shape to be added to the TensorSpecs in specs. |
Returns | |
---|---|
Nested TensorSpecs with outer dimensions added to the shape of input specs. |
Raises | |
---|---|
ValueError
|
if any outer_dims is neither a list nor tuple. |