Module: tfp.layers.util

Utilities for probabilistic layers.

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.

deserialize_function(...): Deserializes the Keras-serialized function.

serialize_function(...): Serializes function for Keras.