Module: tfp.layers

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

class DenseFlipout: Densely-connected layer class with Flipout estimator.

class DenseLocalReparameterization: Densely-connected layer class with local reparameterization estimator.

class DenseReparameterization: Densely-connected layer class with reparameterization estimator.

Functions

default_loc_scale_fn(...): Makes closure which creates loc, scale params from tf.get_variable.

default_mean_field_normal_fn(...): Creates a function to build Normal distributions with trainable params.

default_multivariate_normal_fn(...): Creates multivariate standard Normal distribution.