本文整理汇总了Python中tensorflow.contrib.seq2seq.Helper方法的典型用法代码示例。如果您正苦于以下问题:Python seq2seq.Helper方法的具体用法?Python seq2seq.Helper怎么用?Python seq2seq.Helper使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.contrib.seq2seq
的用法示例。
在下文中一共展示了seq2seq.Helper方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: __init__
# 需要导入模块: from tensorflow.contrib import seq2seq [as 别名]
# 或者: from tensorflow.contrib.seq2seq import Helper [as 别名]
def __init__(self, cell, helper, initial_state, output_layer=None):
"""Initialize BasicVectorDecoder.
Args:
cell: An `RNNCell` instance.
helper: A `Helper` instance.
initial_state: A (possibly nested tuple of...) tensors and TensorArrays.
The initial state of the RNNCell.
output_layer:An instance of `tf.layers.Layer`, i.e., `tf.layers.Dense`.
If not provided, use 1 fc layer.
Raises:
TypeError: if `cell`, `helper` or `output_layer` have an incorrect type.
"""
if not rnn_cell_impl._like_rnncell(cell): # pylint: disable=protected-access
raise TypeError("cell must be an RNNCell, received: %s" % type(cell))
if not isinstance(helper, helper_py.Helper):
raise TypeError("helper must be a Helper, received: %s" % type(helper))
if (output_layer is not None and not isinstance(output_layer, layers_base.Layer)):
raise TypeError(
"output_layer must be a Layer, received: %s" % type(output_layer))
self._cell = cell
self._helper = helper
self._initial_state = initial_state
if output_layer is None:
self._output_layer = layer_core.Dense(2, use_bias=True,
name="stop_predictor")
else:
self._output_layer = output_layer