code: def scaled_identity_operator(shape, dtype):
      init_fn = lambda _: ()
      apply_fn = lambda _: tf.linalg.LinearOperatorScaledIdentity(  # pylint: disable=g-long-lambda
          num_rows=shape[-1], multiplier=tf.constant(1.0, dtype=dtype)
      )
      return init_fn, apply_fn
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import scale_matvec_linear_operator
  from tensorflow_probability.python.experimental.util import deferred_module
  from tensorflow_probability.python.experimental.vi.util import trainable_linear_operators
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 3 for calling a global function with example usage: scaled_identity_operator(shape, dtype) and returns: init_fn,  apply_fn