DEBUG: ensure query batch input and output sizes are the same