Function flatten

GAN/Self_attention_GAN_tensorflow/ops.py:90–91

(x)

        return x

def flatten(x) :
    return tf.layers.flatten(x)

def hw_flatten(x) :
    return tf.reshape(x, shape=[x.shape[0], -1, x.shape[-1]])