Module: tfp.experimental.vi.util

View source on GitHub

Experimental methods and objectives for variational inference.

Functions

build_linear_operator_zeros(...): Build an instance of LinearOperatorZeros.

build_trainable_linear_operator_block(...): Builds a trainable blockwise tf.linalg.LinearOperator.

build_trainable_linear_operator_diag(...): Build a trainable LinearOperatorDiag instance.

build_trainable_linear_operator_full_matrix(...): Build a trainable LinearOperatorFullMatrix instance.

build_trainable_linear_operator_tril(...): Build a trainable LinearOperatorLowerTriangular instance.