将tensorflow 1 contrib转换为tensorflow 2 Keras版本

Question

我正在将我的代码从tf1迁移到tf2，我想我已经设法解决了在tf2上运行它的大多数问题。但是在迁移到兼容tfa.seq2seq.LuongAttention的Tf2时卡住了，tfa.seq2seq.AttentionWrapper已经替换了contrib到v2，但不确定为什么它不能工作。

def _single_cell(num_units, keep_prob, device_str=None):
    single_cell = tf.compat.v1.nn.rnn_cell.GRUCell(num_units)
    if keep_prob < 1.0:
        single_cell = tf.contrib.rnn.DropoutWrapper(cell=single_cell, input_keep_prob=keep_prob)
    # Device Wrapper
    if device_str:
        single_cell = tf.contrib.rnn.DeviceWrapper(single_cell, device_str)
    return single_cell


def create_rnn_cell(num_units, num_layers, keep_prob):
    """Create multi-layer RNN cell."""
    cell_list = []
    for i in range(num_layers):
        single_cell = _single_cell(num_units=num_units, keep_prob=keep_prob)
        cell_list.append(single_cell)
    if len(cell_list) == 1:  # Single layer.
        return cell_list[0]
    else:  # Multi layers
        return tf.compat.v1.nn.rnn_cell.MultiRNNCell(cell_list)

cell = create_rnn_cell(
            num_units=hparams.num_units,
            num_layers=hparams.num_layers,
            keep_prob=hparams.keep_prob)

encoder_outputs, encoder_state = tf.compat.v1.nn.dynamic_rnn(
                cell,
                encoder_emb_inp,
                dtype=dtype,
                sequence_length=self.batch_input.source_sequence_length,
                time_major=self.time_major)

我引用了https://github.com/tensorflow/addons/tree/master/tensorflow_addons/seq2seq，并且能够迁移除这两个函数之外的大部分代码

Answer 1

TensorFlow2.x中的库很少被移动到其他存储库，如插件和操作。

有关库查找here的更多信息，请将tf.contrib.rnn.DropoutWrapper替换为tf.compat.v1.nn.rnn_cell.DropoutWrapper。

有关库查找here的更多信息，请将tf.contrib.rnn.DeviceWrapper替换为tf.compat.v1.nn.rnn_cell.DeviceWrapper。