View source on GitHub |
Adds an outer dimension to the shape of input specs.
tf_agents.specs.tensor_spec.add_outer_dim(
specs, dim=None
)
Used in the notebooks
Used in the tutorials |
---|
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. |