code: def __init__(self):
      super(Flatten, self).__init__()
  
imports: import math
  from types import FunctionType
  import tensorflow
  from rl_coach.architectures import layers
  from rl_coach.architectures.tensorflow_components import utils
label_desc: Use image node_id 1 to create a new Flatten object from inherited base classes: layers with example: obj = Flatten()