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.abstractmethodinitialize( name=None )
Returns (initial_finished, initial_inputs).
next_inputs
@abc.abstractmethodnext_inputs( time, outputs, state, sample_ids, name=None )
Returns (finished, next_inputs, next_state).
sample
@abc.abstractmethodsample( time, outputs, state, name=None )
Returns sample_ids.
View source on GitHub