Module: tf_agents.specs

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

Modules

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

distribution_spec module: Spec definition for tensorflow_probability.Distribution.

tensor_spec module: Utilities related to TensorSpec class.

Classes

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

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

class DistributionSpec: Describes a tfp.distribution.Distribution.