老师，请解释一下这段代码？有点迷糊

来源：9-18 构建解码器（7）

StephenLee147

2019-01-16

if not self.use_beamsearch_decode:

                dod = self.decoder_outputs_decode
                self.decoder_pred_decode = dod.sample_id

                if self.time_major:
                    self.decoder_pred_decode = tf.transpose(
                        self.decoder_pred_decode, (1, 0))

            else:
                self.decoder_pred_decode = \
                    self.decoder_outputs_decode.predicted_ids

                if self.time_major:
                    self.decoder_pred_decode = tf.transpose(
                        self.decoder_pred_decode, (1, 0, 2))

                self.decoder_pred_decode = tf.transpose(
                    self.decoder_pred_decode,
                    perm=[0, 2, 1])
                dod = self.decoder_outputs_decode
                self.beam_prob = dod.beam_search_decoder_output.scores

写回答

1回答