code: def is_flatten(self, input):
      # input is a tensor
      shape = len(input.get_shape().as_list())
      if shape > 2:
          return True
      return False
  
imports: from __future__ import absolute_import, division, print_function
  import logging
  import math
  import tempfile
  import time
  import argparse
  import tensorflow
  from tensorflow.examples.tutorials.mnist import input_data
  import nni
label_desc: Use image node_id 3 for calling the MnistNetwork obj's underlying member method code with example usage: obj.is_flatten(input) and returns: False, True