code: def inflate_by_scaled_identity_fn(scaling_factor):
      """Return function that scales up covariance matrix by `scaling_factor**2`."""
  
      def _inflate_fn(particles):
          particle_means = tf.nest.map_structure(
              lambda x: tf.math.reduce_mean(x, axis=0), particles
          )
          return tf.nest.map_structure(
              lambda x, m: scaling_factor * (x - m) + m,
              particles,
              particle_means,
          )
  
      return _inflate_fn
  
file_path: probability/tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter.py
imports: import collections
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import mvn_linear_operator
  from tensorflow_probability.python.distributions import mvn_low_rank_update_linear_operator_covariance
  from tensorflow_probability.python.distributions import mvn_tril
  from tensorflow_probability.python.internal import distribution_util
  from tensorflow_probability.python.internal import dtype_util
label_desc: Use image node_id 2 for calling a global function with example usage: inflate_by_scaled_identity_fn(scaling_factor) and returns: _inflate_fn, tf