Spec-Zone.ru › TensorFlow 1.15

tf.contrib.legacy_seq2seq.model_with_buckets

Создайте модель последовательность-последовательность с поддержкой разбиения на корзины.

tf.contrib.legacy_seq2seq.model_with_buckets(
    encoder_inputs, decoder_inputs, targets, weights, buckets, seq2seq,
    softmax_loss_function=None, per_example_loss=False, name=None
)

Аргумент seq2seq — функция, определяющая модель последовательность-последовательность, например, seq2seq = lambda x, y: basic_rnn_seq2seq( x, y, rnn_cell.GRUCell(24))

Аргументы
encoder_inputs Список тензоров для подвода данных на кодировщик; первый вход seq2seq.
decoder_inputs Список тензоров для подвода данных на декодер; второй вход seq2seq.
targets Список одномерных тензоров int32 размера пакет (желаемая выходная последовательность).
weights Список одномерных тензоров float для взвешивания целевых значений.
buckets Список пар (размер входа, размер выхода) для каждой корзины.
seq2seq Функция модели последовательность-последовательность; она принимает 2 входа, согласующиеся с encoder_inputs и decoder_inputs, и возвращает пару, состоящую из выходов и состояний (например, как basic_rnn_seq2seq).
softmax_loss_function Функция (метки, логарифмы) -> пакетная потеря, которая должна использоваться вместо стандартной softmax (по умолчанию, если это None). Обратите внимание, что для избежания путаницы, функция должна принимать именованные аргументы.
per_example_loss Булево значение. Если установлено, возвращаемая потеря будет тензором размера пакета потерь для каждой последовательности в пакете. Если не установлено, будет скаляр со средней потерей от всех примеров.
name Необязательное имя для этой операции, по умолчанию "model_with_buckets".
Возвращаемые значения
Кортеж вида (выходы, потери), где: выходы: Выходы для каждой корзины. Его j-ый элемент состоит из списка 2D тензоров. Форма тензоров выходов может быть либо [размер_пакета x размер_выхода], либо [размер_пакета x число_символов_декодера] в зависимости от используемой модели seq2seq. потери: Список скалярных тензоров, представляющих потери для каждой корзины, или, если per_example_loss установлено, список одномерных тензоров float размера пакета.
Исключения
ValueError Если длина encoder_inputs, targets или weights меньше наибольшей (последней) корзины.

© 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/legacy_seq2seq/model_with_buckets

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API