tf.contrib.seq2seq.TrainingHelper
A helper for use during training. Only reads inputs.
Inherits From: Helper
tf.contrib.seq2seq.TrainingHelper( inputs, sequence_length, time_major=False, name=None )
Returned sample_ids are the argmax of the RNN output logits.
Args | |
---|---|
inputs | A (structure of) input tensors. |
sequence_length | An int32 vector tensor. |
time_major | Python bool. Whether the tensors in inputs are time major. If False (default), they are assumed to be batch major. |
name | Name scope for any created operations. |
Raises | |
---|---|
ValueError | if sequence_length is not a 1D tensor. |
Attributes | |
---|---|
batch_size | Batch size of tensor returned by sample . Returns a scalar int32 tensor. |
inputs | |
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 |
sequence_length |
Methods
initialize
initialize( name=None )
Returns (initial_finished, initial_inputs)
.
next_inputs
next_inputs( time, outputs, state, name=None, **unused_kwargs )
next_inputs_fn for TrainingHelper.
sample
sample( time, outputs, name=None, **unused_kwargs )
Returns sample_ids
.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/seq2seq/TrainingHelper