code: def make_batches(N_total, N_batch):
      start = 0
      batches = []
      while start < N_total:
          batches.append(slice(start, start + N_batch))
          start += N_batch
      return batches
  
imports: from __future__ import absolute_import
  from __future__ import print_function
  from builtins import range
  import autograd.numpy
  import autograd.numpy.random
  import autograd.scipy.signal
  from autograd import grad
  import data_mnist
label_desc: Use image node_id 1 for calling a global function with example usage: make_batches(N_total, N_batch) and returns: batches