def METHOD_NAME(self):
    input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size)
    decoder_input_ids = ids_tensor([self.batch_size, self.tgt_seq_length], self.vocab_size)
    input_mask = None
    if self.use_input_mask:
        input_mask = paddle.ones([self.batch_size, self.seq_length], dtype="float32")
    decoder_input_mask = None
    if self.use_input_mask:
        decoder_input_mask = paddle.ones([self.batch_size, self.tgt_seq_length], dtype="float32")
    config = self.get_config()
    return config, input_ids, input_mask, decoder_input_ids, decoder_input_mask