let get_out_size net =
   let neurons = List.hd (List.rev net) in
   Array.length neurons