View source on GitHub |
Interface for implementing sampling in seq2seq decoders.
Helper instances are used by BasicDecoder
.
Attributes | |
---|---|
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 |
Methods
initialize
@abc.abstractmethod
initialize( name=None )
Returns (initial_finished, initial_inputs)
.
next_inputs
@abc.abstractmethod
next_inputs( time, outputs, state, sample_ids, name=None )
Returns (finished, next_inputs, next_state)
.
sample
@abc.abstractmethod
sample( time, outputs, state, name=None )
Returns sample_ids
.