code: def encode_to_shape(inputs, shape, scope):
    
    with tf.variable_scope(scope, reuse=tf.AUTO_REUSE):
      w, h = shape[1], shape[2]
      x = inputs
      x = tfl.flatten(x)
      x = tfl.dense(x, w * h, activation=None, name="enc_dense")
      x = tf.reshape(x, (-1, w, h, 1))
      return x
docs: Encode the given tensor to given image shape.