code: def scaled_I(x, size):
      """Scaled identity matrix
      :math:`x I_{size}`
  
      Parameters
      ------------
      x: float or coercable to float
          Scale to multiply the identity matrix by
      size: int or otherwise coercable to a size
          Dimension of the scaled identity matrix to return
      """
      return tf.linalg.tensor_diag(
          tf.ones([size], dtype=tf.float64) * x
      )
  
file_path: brainiak/brainiak/matnormal/utils.py
imports: import tensorflow
  import tensorflow_probability
  from scipy.stats import norm
  from numpy.linalg import cholesky
  import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: scaled_I(x, size) and returns: tf