beam search using DynamicAttentionWrapper

[yangshao]

Hi, I'm trying to use this beam search code for attentioned to seq2seq model. the type of the decoding cell is <tensorflow.contrib.seq2seq.python.ops.dynamic_attention_wrapper.DynamicAttentionWrapper
and initial state is: DynamicAttentionWrapperState(cell_state=LSTMStateTuple(c=<tf.Tensor 'Test/seq2seq_att/social_bot/tower_0/Decoder/decoder/decoder_1/while/Identity_5:0' shape=(?, 2048) dtype=float32>, h=<tf.Tensor 'Test/seq2seq_att/social_bot/tower_0/Decoder/decoder/decoder_1/while/Identity_6:0' shape=(?, 2048) dtype=float32>), attention=<tf.Tensor 'Test/seq2seq_att/social_bot/tower_0/Decoder/decoder/decoder_1/while/Identity_7:0' shape=(?, 2048) dtype=float32>)

This code seems not work. how should change the code for the new api? Thanks.