Hansu Kim

output_log_probs's shape should be `(gen_length, batch_size*beam_width)`, but id_offset assumes that the shape of tensor is `(max_input_length+gen_legnth, batch_size*beam_width)`. So, I define new index `gen_offset` which assumes that the shape of tensor...