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.

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-04-08.