tfp.experimental.vi.util.build_trainable_linear_operator_diag

Build a trainable LinearOperatorDiag instance.

tfp.experimental.vi.util.build_trainable_linear_operator_diag(
    *args, seed=None, **kwargs
)

shape Shape of the LinearOperator, equal to [b0, ..., bn, d], where b0...bn are batch dimensions and d is the length of the diagonal.
scale_initializer Variables are initialized with samples from Normal(0, scale_initializer).
diag_bijector Bijector to apply to the diagonal of the operator.
dtype tf.dtype of the LinearOperator.
name str, name for tf.name_scope. seed: PRNG seed; see tfp.random.sanitize_seed for details.

instance instance parameterized by trainable tf.Variables.