Module: tf_agents.specs.array_spec

View source on GitHub

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

Classes

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

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

Functions

add_outer_dims_nest(...)

check_arrays_nest(...): Check that the arrays conform to the spec.

is_bounded(...)

is_continuous(...)

is_discrete(...)

sample_bounded_spec(...): Samples the given bounded spec.

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

update_spec_dtype(...): Returns a copy of the given spec with the new dtype.

update_spec_shape(...): Returns a copy of the given spec with the new shape.