View source on GitHub
|
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.