![]() |
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.