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

Build a trainable LinearOperatorDiag instance.

shape Shape of the LinearOperator, equal to [b0, ..., bn, d], where 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.
seed Python integer to seed the random number generator.
name str, name for tf.name_scope.

operator Trainable instance of tf.linalg.LinearOperatorDiag.