python - 张量流中的 Seq2Seq 训练助手模块

Question

在最新的 tf 2.0 更新中，tensorflow 从框架中删除了 contrib 模块。以为他们已经补偿了 tf.compat.v1 中的大部分功能，但我找不到 tf.contrib.seq2seq.BasicDecoder 、 tf.contrib.seq2seq.dynamic_decode 、 tf.contrib 等功能的替代品。 seq2seq.GreedyEmbeddingHelper 和 tf.contrib.seq2seq.TrainingHelper。如何在我的模型中使用这些功能？

标签： pythonkerastensorflow2.0

Answer 1

查看 TensorFlow-Addons：

https://github.com/tensorflow/addons ,

https://www.tensorflow.org/addons/api_docs/python/tfa/seq2seq?hl=en

您正在寻找的功能都在那里。

pip install tensorflow-addons

然后：

import tensorflow as tf
import tensorflow_addons as tfa