ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.contrib.seq2seq.Helper

View source on GitHub

Interface for implementing sampling in seq2seq decoders.

Helper instances are used by BasicDecoder.

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

View source

Returns (initial_finished, initial_inputs).

next_inputs

View source

Returns (finished, next_inputs, next_state).

sample

View source

Returns sample_ids.