View source on GitHub |
Create zero tensors for a given spec.
tf_agents.specs.zero_spec_nest(
specs, outer_dims=None
)
Args | |
---|---|
specs
|
A nest of TensorSpec .
|
outer_dims
|
An optional list of constants or Tensor specifying outer
dimensions to add to the spec shape before sampling.
|
Returns | |
---|---|
A nest of zero tensors matching specs , with the optional outer
dimensions added.
|
Raises | |
---|---|
TypeError
|
If specs is an unknown type.
|
NotImplementedError
|
If specs contains non-dense tensor specs.
|