let one_letter_cross_entropy dist y =
  lg (assoc y dist)