def shake_shake_eval(xa, xb):
  """Shake-shake regularization in testing mode.

  Args:
    xa: Input, branch A.
    xb: Input, branch B.

  Returns:
    Mix of input branches.
  """
  # Blend between inputs A and B 50%-50%.
  return (xa + xb) * 0.5