tf.contrib.seq2seq.InferenceHelper
Помощник для использования во время вывода с пользовательской функцией выборки.
Наследуется от: Helper
tf.contrib.seq2seq.InferenceHelper(
sample_fn, sample_shape, sample_dtype, start_inputs, end_fn, next_inputs_fn=None
)
| Аргументы | |
|---|---|
sample_fn | Вызываемый объект, принимающий outputs и генерирующий тензор sample_ids. |
sample_shape | Список целых чисел или 1-мерный тензор типа int32, форма каждого образца в пакетe, возвращаемом sample_fn. |
sample_dtype | Тип данных образца, возвращаемого sample_fn. |
start_inputs | Начальный пакет входных данных. |
end_fn | Вызываемый объект, принимающий sample_ids и генерирующий вектор bool, с формой [batch_size], указывающий, является ли каждый образец маркером окончания. |
next_inputs_fn | (Необязательно) Вызываемый объект, принимающий sample_ids и возвращающий следующий пакет входных данных. Если не предоставлен, sample_ids используется в качестве следующего пакета входных данных. |
| Атрибуты | |
|---|---|
batch_size | Размер пакета тензора, возвращаемого sample. Возвращает скалярный тензор int32. |
sample_ids_dtype | Тип данных тензора, возвращаемого sample. Возвращает тип данных. |
sample_ids_shape | Форма тензора, возвращаемого sample, исключая размер пакета. Возвращает |
Методы
initialize
initialize(
name=None
)
Возвращает (initial_finished, initial_inputs).
next_inputs
next_inputs(
time, outputs, state, sample_ids, name=None
)
Возвращает (finished, next_inputs, next_state).
sample
sample(
time, outputs, state, name=None
)
Возвращает 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/InferenceHelper