tf.linalg.LinearOperatorDiag

LinearOperator acting like a [batch] square diagonal matrix.

Inherits From: LinearOperator, Module

Used in the notebooks

Used in the tutorials