code: def _make_eye(shape):
      if len(shape) == 2:
          n = shape[0]
          return numpy.eye(n, dtype=numpy.float32)
      m = shape[0]
      n = shape[1]
      return numpy.array([numpy.eye(n, dtype=numpy.float32)] * m)
  
imports: import unittest
  import numpy
  import chainer
  from chainer.backends import cuda
  from chainer import functions
  from chainer import testing
  from chainer.testing import attr
  from chainer.utils import type_check
label_desc: Use image node_id 2 for calling a global function with example usage: _make_eye(shape) and returns: numpy, numpy