Что эквивалентно tf.contrib.seq2seq.prepare_attention в TensorFlow 2 - PullRequest
Новая
       66

Что эквивалентно tf.contrib.seq2seq.prepare_attention в TensorFlow 2

1 голос
/ 12 июля 2020

Недавно я работал над некоторым кодом, написанным на tenorflow 1.0.1, и хочу сделать его доступным на tenorflow 2. Я не очень знаком с seq2seq. Большое спасибо. 

(attention_keys,
attention_values,
attention_score_fn,
attention_construct_fn) = tf.contrib.seq2seq.prepare_attention(
    attention_states=attention_states,
    attention_option="bahdanau",
    num_units=self.decoder_hidden_units,
)

1 Ответ

0 голосов
/ 15 июля 2020

Согласно этому Комментарий Github , tf.contrib.seq2seq.prepare_attention() переименован в tf.contrib.seq2seq.DynamicAttentionWrapper.

Согласно этому Github Tensorflow Commit , DynamicAttentionWrapper был переименован на AttentionWrapper.

Итак, в 1.15 функция, эквивалентная tf.contrib.seq2seq.prepare_attention(), равна tf.contrib.seq2seq.AttentionWrapper.

Функция, эквивалентная tf.contrib.seq2seq.AttentionWrapper в Tensorflow 2.x is tfa.seq2seq.AttentionWrapper.

Дополнительную информацию см. В Документации по Tensorflow .

...