tf_agents.specs.zero_spec_nest

Create zero tensors for a given spec.

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.

A nest of zero tensors matching specs, with the optional outer dimensions added.

TypeError If specs is an unknown type.
NotImplementedError If specs contains non-dense tensor specs.