def flatten_sum()

in tfops.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def flatten_sum(logps):
    if len(logps.get_shape()) == 2:
        return tf.reduce_sum(logps, [1])
    elif len(logps.get_shape()) == 4:
        return tf.reduce_sum(logps, [1, 2, 3])
    else:
        raise Exception()