tensorflow更新后如何导入BeamSearchDecoder？

Question

from tensorflow.contrib.seq2seq import BeamSearchDecoder as beam_search_decoder

给予

ModuleNotFoundError: No module named 'tensorflow.contrib'

Tensorflow的最新版本不包含contrib，因此如何使用该函数

Answer 1

BeamSearchDecoder现在是Tensorflow Addons的一部分，应该通过pip单独安装：

pip install tensorflow-addons

之后，您可以在此处尝试使用它：

from tensorflow_addons.seq2seq import BeamSearchDecoder as beam_search_decoder

或者直接按照{a3}和{a4}中的建议：

import tensorflow_addons as tfa
tfa.seq2seq.BeamSearchDecoder(...)