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