def lrelu_sq(x):
    """
    Concatenates lrelu and square
    """
    dim = len(x.get_shape()) - 1
    return tf.concat(dim, [lrelu(x), tf.minimum(tf.abs(x), tf.square(x))])
Concatenates lrelu and square