Google I/O returns May 18-20! Reserve space and build your schedule Register now

tfp.experimental.vi.util.build_trainable_linear_operator_full_matrix

Build a trainable LinearOperatorFullMatrix instance.

tfp.experimental.vi.util.build_trainable_linear_operator_full_matrix(
    shape, scale_initializer=0.01, dtype=None, seed=None, name=None
)

shape Shape of the LinearOperator, equal to [b0, ..., bn, h, w], where b0...bn are batch dimensions h and w are the height and width of the matrix represented by the LinearOperator.
scale_initializer Variables are initialized with samples from Normal(0, scale_initializer).
dtype tf.dtype of the LinearOperator.
seed Python integer to seed the random number generator.
name str, name for tf.name_scope.

operator Trainable instance of tf.linalg.LinearOperatorFullMatrix.