Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.layers.util

Utilities for probabilistic layers.

Defined in python/layers/


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.