beam search的问题

看了一下BeamSearchSampler的源码,似乎是不支持有attention的decoder。那么我改一下__call__方法,把encdoer的output传入是否可行?

现在的接口是支持带有attention的decoder的。例子里的Transformer就是。可以把encoding得到的memory传到states里面。

多谢指教,所以state其实可以是一个ndarray组成的list?

是的,states可以是一个list,只需要保证输入的states和输出的states是匹配的就行了,就能使用beam search.

了解了,十分感谢 :sunglasses: