gpflux.types

Types used within GPflux (for static type-checking).

Module Contents

unwrap_dist(dist: tfp.distributions.Distribution) → tfp.distributions.Distribution

Unwrap the given distribution, if it is wrapped in a _TensorCoercible.

ShapeType

Union of valid types for describing the shape of a tf.Tensor(-like) object

ObservationType

Type for the [inputs, targets] list used by LayerWithObservations