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.