code: def eye(
      num_rows,
      num_columns=None,
      batch_shape=None,
      dtype=dtypes.float32,
      name=None,
  ):
      """Construct an identity matrix, or a batch of matrices.
  
      See also `tf.ones`, `tf.zeros`, `tf.fill`, `tf.one_hot`.
  
      ```python
      # Construct one identity matrix.
      tf.eye(2)
      ==> [[1., 0.],
           [0., 1.]]
  
      # Construct a batch of 3 identity matrices, each 2 x 2.
      # batch_identity[i, :, :] is a 2 x 2 identity matrix, i = 0, 1, 2.
      batch_identity = tf.eye(2, batch_shape=[3])
  
      # Construct one 2 x 3 "identity" matrix
      tf.eye(2, num_columns=3)
      ==> [[ 1.,  0.,  0.],
           [ 0.,  1.,  0.]]
      ```
  
      Args:
        num_rows: Non-negative `int32` scalar `Tensor` giving the number of rows
          in each batch matrix.
        num_columns: Optional non-negative `int32` scalar `Tensor` giving the number
          of columns in each batch matrix.  Defaults to `num_rows`.
        batch_shape:  A list or tuple of Python integers or a 1-D `int32` `Tensor`.
          If provided, the returned `Tensor` will have leading batch dimensions of
          this shape.
        dtype:  The type of an element in the resulting `Tensor`
        name:  A name for this `Op`.  Defaults to "eye".
  
      Returns:
        A `Tensor` of shape `batch_shape + [num_rows, num_columns]`
      """
      return linalg_ops_impl.eye(
          num_rows,
          num_columns=num_columns,
          batch_shape=batch_shape,
          dtype=dtype,
          name=name,
      )
  
imports: import numpy
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import cond
  from tensorflow.python.ops import gen_array_ops
  from tensorflow.python.ops import gen_linalg_ops
  from tensorflow.python.ops import linalg_ops_impl
  from tensorflow.python.ops import map_fn
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops.gen_linalg_ops import *
  from tensorflow.python.util import deprecation
  from tensorflow.python.util import dispatch
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 4 for calling a global function with example usage: eye(num_rows, num_columns, batch_shape, dtype, name) and returns: linalg_ops_impl