def merge_states()

in src/model.py [0:0]

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

def merge_states(x):
    """Smash the last two dimensions of x into a single dimension."""
    *start, a, b = shape_list(x)
    return tf.reshape(x, start + [a*b])