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

tf_agents.specs.tensor_spec.sample_spec_nest

View source on GitHub

Samples the given nest of specs.

tf_agents.specs.tensor_spec.sample_spec_nest(
    structure, seed=None, outer_dims=()
)

Args:

  • structure: A nest of TensorSpec.
  • seed: A seed used for sampling ops
  • outer_dims: An optional Tensor specifying outer dimensions to add to the spec shape before sampling.

Returns:

A nest of sampled values following the ArraySpec definition.

Raises:

  • TypeError: If spec is an unknown type.
  • NotImplementedError: If outer_dims is not statically known but nest contains a SparseTensorSpec.