def encoder(self):
self.encoder_outputs, self.encoder_state = tf.nn.dynamic_rnn(
cell=tf.contrib.rnn.GRUCell(self.encoder_hidden_size), inputs=self.inputs, sequence_length=self.encoder_length,
dtype=tf.float32, scope="rnn", time_major=False) # tf.variable_scope("rnn")
self.encoder_outputs = tf.contrib.layers.linear(self.encoder_state, self.num_decoder_symbols)
self.encoder_softmax_outputs = tf.nn.softmax(self.encoder_outputs)
return #encoder_outputs, encoder_state
评论列表
文章目录