View source on GitHub
|
Experimental Joint Distribution Layers library.
Classes
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.
Functions
make_conv2d_layer_class(...): Creates a Conv2DLayer class.
make_lambda_layer_class(...): Creates a LambdaLayer class.