tf.nn.seq2seq 函数 新版本调整

Tensorflow 函数调整:

tf.nn.seq2seq.sequence_loss_by_example(
            [logits],   # output [batch*numsteps, vocab_size]
            [tf.reshape(self._targets, [-1])],  # target, [batch_size, num_steps]
            [tf.ones([batch_size * num_steps], dtype=data_type())]) # weight

改为:

tf.contrib.legacy_seq2seq.sequence_loss_by_example(
            [logits],   # output [batch*numsteps, vocab_size]
            [tf.reshape(self._targets, [-1])],  # target, [batch_size, num_steps]
            [tf.ones([batch_size * num_steps], dtype=data_type())]) # weight

猜你喜欢

转载自blog.csdn.net/zgqcs55/article/details/84639005