-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Description
inference_decoder_output, _, _ = tf.contrib.seq2seq.dynamic_decode(inference_decoder, maximum_iterations=self.max_decode_step) self.decoder_pred_decode = inference_decoder_output.predicted_ids self.decoder_pred_decode = tf.transpose(self.decoder_pred_decode,perm=[0, 2, 1]
请教一下,这里predicted_ids的维度应该是[batch_size, decoder_targets_length, beam_size],为什么最后要做一次transpose呢?
Metadata
Metadata
Assignees
Labels
No labels