code: def shakeshake2_py(x, y, equal=False, individual=False):
    
    if equal:
      alpha = 0.5
    elif individual:
      alpha = tf.random_uniform(tf.get_shape(x)[:1])
    else:
      alpha = tf.random_uniform([])
  
    return alpha * x + (1.0 - alpha) * y
docs: The shake-shake sum of 2 tensors, python version.