Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.specs.tensor_spec.zero_spec_nest

View source on GitHub

Create zero tensors for a given spec.

tf_agents.specs.tensor_spec.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.