tf.linalg.LinearOperatorScaledIdentity