tf.contrib.seq2seq.CustomHelper

Class CustomHelper

Inherits From: Helper

Defined in tensorflow/contrib/seq2seq/python/ops/helper.py.

See the guide: Seq2seq Library (contrib) > Dynamic Decoding

Base abstract class that allows the user to customize sampling.

Properties

batch_size

sample_ids_dtype

sample_ids_shape

Methods

__init__

__init__(
    initialize_fn,
    sample_fn,
    next_inputs_fn,
    sample_ids_shape=None,
    sample_ids_dtype=None
)

Initializer.

Args:

  • initialize_fn: callable that returns (finished, next_inputs) for the first iteration.
  • sample_fn: callable that takes (time, outputs, state) and emits tensor sample_ids.
  • next_inputs_fn: callable that takes (time, outputs, state, sample_ids) and emits (finished, next_inputs, next_state).
  • sample_ids_shape: Either a list of integers, or a 1-D Tensor of type int32, the shape of each value in the sample_ids batch. Defaults to a scalar.
  • sample_ids_dtype: The dtype of the sample_ids tensor. Defaults to int32.

initialize

initialize(name=None)

next_inputs

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

sample

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