code: def build_cell(units, cell_type="lstm", num_layers=1):
      if num_layers > 1:
          cell = rnn.MultiRNNCell(
              [
                  build_cell(units, cell_type, 1)
                  for _ in range(num_layers)
              ]
          )
      else:
          if cell_type == "lstm":
              cell = rnn.LSTMCell(units)
          elif cell_type == "gru":
              cell = rnn.GRUCell(units)
          else:
              raise ValueError("Do not support %s" % cell_type)
      return cell
  
imports: import tensorflow
  from tensorflow.contrib import rnn
  from tensorflow.python.ops import array_ops
  from ..models.basic_model import BasicModel
  import numpy
  import os
label_desc: Use image node_id 2 for calling a global function with example usage: build_cell(units, cell_type, num_layers) and returns: cell