View source on GitHub |
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
.