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
python tensorflow input shape
Encode the given tensor to given image shape .
def encode_to_shape(inputs, shape, scope):
  """Encode the given tensor to given image shape."""
  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