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
)

Args
`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.

Returns
`instance`	instance parameterized by trainable `tf.Variable`s.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2023-11-21 UTC.

Args