Attend the Women in ML Symposium on December 7 Register now

Module: tf_agents.specs

Stay organized with collections Save and categorize content based on your preferences.

Module for numpy array and tf.Tensor shape and dtype specifications.


array_spec module: A class to describe the shape and dtype of numpy arrays.

bandit_spec_utils module: Bandit related tensor spec utilities.

distribution_spec module: Spec definition for tensorflow_probability.Distribution.

tensor_spec module: Utilities related to TensorSpec class.


class ArraySpec: Describes a numpy array or scalar shape and dtype.

class BoundedArraySpec: An ArraySpec that specifies minimum and maximum values.


from_spec(...): Maps the given spec into corresponding TensorSpecs keeping bounds.




sample_spec_nest(...): Samples the given nest of specs.

zero_spec_nest(...): Create zero tensors for a given spec.