|View source on GitHub|
A Sampler for use during training.
tfa.seq2seq.TrainingSampler( time_major: bool = False )
Used in the notebooks
|Used in the tutorials|
Only reads inputs.
Returned sample_ids are the argmax of the RNN output logits.
Python bool. Whether the tensors in
Batch size of tensor returned by
Returns a scalar int32 tensor. The return value might not available before the invocation of initialize(), in this case, ValueError is raised.
DType of tensor returned by
Returns a DType. The return value might not available before the invocation of initialize().
Shape of tensor returned by
initialize( inputs, sequence_length=None, mask=None )
Initialize the TrainSampler.
||A (structure of) input tensors.|
||An int32 vector tensor.|
||A boolean 2D tensor.|
|(finished, next_inputs), a tuple of two items. The first item is a boolean vector to indicate whether the item in the batch has finished. The second item is the first slide of input data based on the timestep dimension (usually the second dim of the input).|
next_inputs( time, outputs, state, sample_ids )
(finished, next_inputs, next_state).
sample( time, outputs, state )