(encoder_inputs, decoder_inputs, do_decode)
| 33 | |
| 34 | # 定义函数 |
| 35 | def seq2seq_f(encoder_inputs, decoder_inputs, do_decode): |
| 36 | return tf.nn.seq2seq.embedding_rnn_seq2seq( |
| 37 | encoder_inputs, decoder_inputs, cell, |
| 38 | num_encoder_symbols=vocab_size, |
| 39 | num_decoder_symbols=vocab_size, |
| 40 | embedding_size=num_units, |
| 41 | feed_previous=do_decode) |
| 42 | |
| 43 | # 初始化训练用的变量,如果是多个层,权重共享 |
| 44 | encoder_inputs = [] |
no outgoing calls
no test coverage detected