View source on GitHub |
Convolutional variational layers.
Classes
class Convolution1DFlipout
: 1D convolution layer (e.g. temporal convolution) with Flipout.
class Convolution1DReparameterization
: 1D convolution layer (e.g. temporal convolution).
class Convolution2DFlipout
: 2D convolution layer (e.g. spatial convolution over images) with Flipout.
class Convolution2DReparameterization
: 2D convolution layer (e.g. spatial convolution over images).
class Convolution3DFlipout
: 3D convolution layer (e.g. spatial convolution over volumes) with Flipout.
class Convolution3DReparameterization
: 3D convolution layer (e.g. spatial convolution over volumes).