Class InferenceHelper
Inherits From: Helper
Defined in tensorflow/contrib/seq2seq/python/ops/helper.py
.
A helper to use during inference with a custom sampling function.
__init__
__init__(
sample_fn,
sample_shape,
sample_dtype,
start_inputs,
end_fn,
next_inputs_fn=None
)
Initializer.
Args:
sample_fn
: A callable that takesoutputs
and emits tensorsample_ids
.sample_shape
: Either a list of integers, or a 1-D Tensor of typeint32
, the shape of the each sample in the batch returned bysample_fn
.sample_dtype
: the dtype of the sample returned bysample_fn
.start_inputs
: The initial batch of inputs.end_fn
: A callable that takessample_ids
and emits abool
vector shaped[batch_size]
indicating whether each sample is an end token.next_inputs_fn
: (Optional) A callable that takessample_ids
and returns the next batch of inputs. If not provided,sample_ids
is used as the next batch of inputs.
Properties
batch_size
Batch size of tensor returned by sample
.
Returns a scalar int32 tensor.
sample_ids_dtype
DType of tensor returned by sample
.
Returns a DType.
sample_ids_shape
Shape of tensor returned by sample
, excluding the batch dimension.
Returns a TensorShape
.
Methods
tf.contrib.seq2seq.InferenceHelper.initialize
initialize(name=None)
Returns (initial_finished, initial_inputs)
.
tf.contrib.seq2seq.InferenceHelper.next_inputs
next_inputs(
time,
outputs,
state,
sample_ids,
name=None
)
Returns (finished, next_inputs, next_state)
.
tf.contrib.seq2seq.InferenceHelper.sample
sample(
time,
outputs,
state,
name=None
)
Returns sample_ids
.