Could you plz fix the issue about beam search decoder with copynet?

[bb9696aa]

Hi there, the code is not working when I set the parameters beam_width and num_translations_per_input to be ≥ 1.

E.g., when I set the beam_width=9, batch_size=32, the error information is shown as follows:

InvalidArgumentError (see above for traceback): Incompatible shapes: [288,1] vs. [32,11]
	 [[Node: dynamic_seq2seq/decoder/decoder/while/BeamSearchDecoderStep/Equal = Equal[T=DT_INT32, _device="/job:localhost/replica:0/task:0/device:CPU:0"](dynamic_seq2seq/decoder/decoder/while/BeamSearchDecoderStep/ExpandDims, dynamic_seq2seq/decoder/decoder/while/BeamSearchDecoderStep/Equal/Enter)]]

I dont have any ideas about how to fix this problem, any replies will be appreciated! Thank you!