tf.contrib.seq2seq.InferenceHelper

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 takes outputs and emits tensor sample_ids.
  • sample_shape: Either a list of integers, or a 1-D Tensor of type int32, the shape of the each sample in the batch returned by sample_fn.
  • sample_dtype: the dtype of the sample returned by sample_fn.
  • start_inputs: The initial batch of inputs.
  • end_fn: A callable that takes sample_ids and emits a bool vector shaped [batch_size] indicating whether each sample is an end token.
  • next_inputs_fn: (Optional) A callable that takes sample_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

initialize

initialize(name=None)

Returns (initial_finished, initial_inputs).

next_inputs

next_inputs(
    time,
    outputs,
    state,
    sample_ids,
    name=None
)

Returns (finished, next_inputs, next_state).

sample

sample(
    time,
    outputs,
    state,
    name=None
)

Returns sample_ids.