code: def fake_mnist_data():
      def gen_examples(num_examples):
          x = np.array(
              np.random.randn(num_examples, 784),
              copy=False,
              dtype=np.float32,
          )
          y = np.zeros((num_examples, 10), dtype=np.float32)
          y[:][0] = 1.0
          return (x, y)
  
      return (gen_examples(100), gen_examples(10), gen_examples(10))
  
file_path: trax/trax/tf_numpy/examples/mnist/train_test.py
imports: import mock
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy.examples.mnist import dataset
  from trax.tf_numpy.examples.mnist import train
label_desc: Use image node_id 1 for calling a global function with example usage: fake_mnist_data() and returns: gen_examples,  gen_examples,  gen_examples, x,  y