Module: tfp.layers.conv_variational

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).