tf.contrib.seq2seq.GreedyEmbeddingHelper

Помощник для использования во время инференции.

Наследуется от: Helper

tf.contrib.seq2seq.GreedyEmbeddingHelper(
    embedding, start_tokens, end_token
)

Использует argmax выходного значения (обрабатывается как logits) и передает результат через слой встраивания, чтобы получить следующий вход.

Аргументы
`embedding`	Вызываемый объект, который принимает векторный тензор `ids` (argmax id) или аргумент `params` для `embedding_lookup`. Возвращаемый тензор будет передан в декодер в качестве входных данных.
`start_tokens`	Вектор с `[batch_size]` формой, начальные токены.
`end_token`	Скаляр, токен, обозначающий конец декодирования.

Исключения
`ValueError`	если `start_tokens` не является одномерным тензором или `end_token` не является скаляром.

Атрибуты
`batch_size`	Размер пакета тензора, возвращаемого `sample`. Возвращает скалярный тензор int32.
`sample_ids_dtype`	Тип данных тензора, возвращаемого `sample`. Возвращает тип данных.
`sample_ids_shape`	Форма тензора, возвращаемого `sample`, за исключением размерности пакета. Возвращает `TensorShape`.

Методы

initialize(
    name=None
)

Возвращает (initial_finished, initial_inputs).

next_inputs(
    time, outputs, state, sample_ids, name=None
)

next_inputs_fn для GreedyEmbeddingHelper.

sample(
    time, outputs, state, name=None
)

sample для GreedyEmbeddingHelper.

© 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/GreedyEmbeddingHelper