提交 b23a23c9 编写于 作者: Z zhanghaichao

fixed error in beam_search example and documents

上级 29026f9f
...@@ -3563,9 +3563,15 @@ def beam_search(step, ...@@ -3563,9 +3563,15 @@ def beam_search(step,
simple_rnn += last_time_step_output simple_rnn += last_time_step_output
return simple_rnn return simple_rnn
generated_word_embedding = GeneratedInput(
size=target_dictionary_dim,
embedding_name="target_language_embedding",
embedding_size=word_vector_dim)
beam_gen = beam_search(name="decoder", beam_gen = beam_search(name="decoder",
step=rnn_step, step=rnn_step,
input=[StaticInput(encoder_last)], input=[StaticInput(encoder_last),
generated_word_embedding],
bos_id=0, bos_id=0,
eos_id=1, eos_id=1,
beam_size=5) beam_size=5)
...@@ -3584,7 +3590,8 @@ def beam_search(step, ...@@ -3584,7 +3590,8 @@ def beam_search(step,
You can refer to the first parameter of recurrent_group, or You can refer to the first parameter of recurrent_group, or
demo/seqToseq/seqToseq_net.py for more details. demo/seqToseq/seqToseq_net.py for more details.
:type step: callable :type step: callable
:param input: Input data for the recurrent unit :param input: Input data for the recurrent unit, which should include the
previously generated words as a GeneratedInput object.
:type input: list :type input: list
:param bos_id: Index of the start symbol in the dictionary. The start symbol :param bos_id: Index of the start symbol in the dictionary. The start symbol
is a special token for NLP task, which indicates the is a special token for NLP task, which indicates the
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册