tf.contrib.seq2seq.SampleEmbeddingHelper

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

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

tf.contrib.seq2seq.SampleEmbeddingHelper(
    embedding, start_tokens, end_token, softmax_temperature=None, seed=None
)

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

Аргументы
`embedding`	Функция, которая принимает векторный тензор `ids` (argmax ids) или аргумент `params` для `embedding_lookup`. Возвращаемый тензор будет передан в декодер в качестве входных данных.
`start_tokens`	`int32` вектор, имеющий форму `[batch_size]`, начальные токены.
`end_token`	`int32` скаляр, токен, обозначающий конец декодирования.
`softmax_temperature`	(Необязательно) `float32` скаляр, значение, на которое делятся логиты перед вычислением softmax. Более высокие значения (выше 1.0) приводят к более случайным выборкам, а более низкие значения приближают распределение выборки к argmax. Должно быть строго больше 0. По умолчанию 1.0.
`seed`	(Необязательно) Семена для выборки.

Исключения
`ValueError`	если `start_tokens` не является тензором 1D или `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 для SampleEmbeddingHelper.

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