如何在TensorFlow中实现自定义波束搜索?

时间:2017-02-02 00:10:54

标签: tensorflow beam-search

原始Google's SmartReply paper

  

我们的搜索如下进行。首先,R的元素(所有可能的响应的集合)被组织成一个特里。然后,我们进行从左到右的光束搜索,但只保留出现在trie中的假设。该搜索过程对于波束大小b和最大响应长度l具有复杂度O(bl)。 b和l通常都在10-30范围内,因此这种方法大大缩短了查找最佳响应的时间,是使该系统可部署的关键因素。

我想实现类似的东西。一些问题:

  1. 如何在TF中的trie数据结构中表示所有可能响应(R)的集合?
  2. 如何修改当前波束搜索以仅保留R?
  3. 中可用的假设

0 个答案:

没有答案