def get_start_input(self, batch_size): start_input = melt.constants(self.start_id, [batch_size], tf.int32) return start_input
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