code: def _eye(
      num_rows,
      num_columns=None,
      batch_shape=None,
      dtype=np.float32,
      name=None,
  ):  # pylint: disable=unused-argument
      dt = utils.numpy_dtype(dtype)
      x = np.eye(num_rows, num_columns).astype(dt)
      if batch_shape is not None:
          x = x * np.ones(tuple(batch_shape) + (1, 1)).astype(dt)
      return x
  
imports: import collections
  import functools
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
label_desc: Use image node_id 7 for calling a global function with example usage: _eye(num_rows, num_columns, batch_shape, dtype, name) and returns: x