波束搜索解码器Tensorflow 2.0

我希望在Tensorflow 2.0 alpha中实现一个具有注意力和波束搜索的序列到序列神经网络。虽然他们网站上的教程非常有用，但我很难找到实现beam搜索的最佳方法，因为contrib库已被弃用-有人能给我指出正确的方向吗

我尝试使用TF2.0s升级脚本将tensorflow 1.X beam搜索升级到2.0，但它不支持contrib库

这就是beam搜索代码查找1.x的方式

decoder=tf.contrib.seq2seq.BeamSearchDecoder(
单元=解码器单元，
嵌入=自嵌入，
开始标记=tf.fill（[self.batch_size]，tf.constant（2）），
结束标记=tf.常数（3），
初始状态=初始状态，
波束宽度=自波束宽度，
输出层=自投影层
)
输出，u，u=tf.contrib.seq2seq.dynamic_解码(
解码器，输出\u时间\u主要=真，最大\u迭代次数=摘要\u最大\u长度，范围=解码器\u范围）
self.prediction=tf.transpose（outputs.predicted_id，perm=[1,2,0]）