コード例 #1
0
ファイル: rnn_decoder.py プロジェクト: liuyuemaicha/hasky 
 def get_start_input(self, batch_size):
     start_input = melt.constants(self.start_id, [batch_size], tf.int32)
     return start_input
コード例 #2
0
 def get_start_input(self, encoder_output):
   batch_size = tf.shape(encoder_output)[0]
   start_input = melt.constants(self.start_id, [batch_size], tf.int32)
   return start_input