import tensorflow as tf

                                    hyper_decoder=hyper_decoder, 
                                    estimator=entropy_bottleneck)
  status = checkpoint.restore(tf.train.latest_checkpoint(ckpt_dir))

  x = tf.convert_to_tensor(x_color, "float32")
  x_coori = tf.convert_to_tensor(x_coori, "float32")

  def loop_analysis(element):
    x = tf.expand_dims(element[0], 0)
    x_coori = tf.expand_dims(element[1], 0)
    y = analysis_transform(x_coori,x)
    return tf.squeeze(y,axis=0)

  element = [x,x_coori]
  ys = tf.map_fn(loop_analysis, element, dtype=tf.float32, parallel_iterations=1, back_prop=False)
  print("Analysis Transform")

  def loop_hyper_encoder(y):
