BasicDecoder
BasicDecoder是 的子类,它组装了 RNNCell 和 的实例作为成员,其中DecodeHelper用来实现不同的解码策略。它依次执行以下步骤来完成单步解码:
执行
cell_outputs, cell_states = cell.call(inputs, states)
以获取输出和新的状态。执行
sample_ids = helper.sample(time, cell_outputs, cell_states)
以采样id并将其作为当前步的解码结果。执行
finished, next_inputs, next_states = helper.next_inputs(time, cell_outputs, cell_states, sample_ids)
以产生下一解码步的结束标识、输入和状态。
参数:
cell (RNNCell) - RNNCell的实例或者具有相同接口定义的对象。
output_fn (可选) - 处理cell输出的接口,在采样之前使用。默认值None。
示例代码
初始化,包括helper的初始化和cell的初始化,cell初始化直接使用 initial_cell_states
作为结果。
参数:
- initial_cell_states (Variable) - 单个tensor变量或tensor变量组成的嵌套结构。这是由调用者 dynamic_decode 提供的参数。
返回:(initial_inputs, initial_states, finished)
的三元组。 initial_inputs, initial_states
均是单个tensor变量或tensor变量组成的嵌套结构, finished
是bool类型的tensor。 initial_inputs, finished
与 helper.initialize()
返回的内容相同; initial_states
与输入参数中的 initial_cell_states
的相同。
返回类型:tuple
class ( cell_outputs, sample_ids )
step()
的返回值中 outputs
使用的数据结构,是一个由 cell_outputs
和 sample_ids
这两个字段构成的命名元组。
step
( time, inputs, states, \*kwargs* )
按照以下步骤执行单步解码:
执行
cell_outputs, cell_states = cell.call(inputs, states)
以获取输出和新的状态。
参数:
time (Variable) - 调用者提供的形状为[1]的tensor,表示当前解码的时间步长。其数据类型为int64。
inputs (Variable) - tensor变量。在第一个解码时间步时与由
initialize()
返回的initial_inputs
相同,其他时间步与由step()
返回的next_inputs
相同。states (Variable) - tensor变量的结构。在第一个解码时间步时与 返回的
initial_states
相同,其他时间步与由step()
返回的next_states
相同。kwargs - 附加的关键字参数,由调用者 提供。
返回: (outputs, next_states, next_inputs, finished)
的四元组。 outputs
是包含 cell_outputs
和 sample_ids
两个字段的命名元组,其中 cell_outputs
是 cell.call()
的结果, sample_ids
是 helper.sample()
的结果; next_states, next_inputs
分别和输入参数中的 有相同的的结构、形状和数据类型; finished
是一个bool类型的tensor,形状是
。