Attend the Women in ML Symposium on December 7 Register now

Module: tfp.experimental.joint_distribution_layers

Stay organized with collections Save and categorize content based on your preferences.

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.