ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.experimental.joint_distribution_layers

Experimental Joint Distribution Layers library.


class Affine: Distribution over affine transformations.

class AffineLayer: Affine layer.

class Conv2D: Distribution over convolutions.

class Lambda: Distribution over callable atoms.

class Sequential: Sequential layer distribution.

class SequentialLayer: Sequential application of multiple layers.


make_conv2d_layer_class(...): Creates a Conv2DLayer class.

make_lambda_layer_class(...): Creates a LambdaLayer class.