Method forward_decoding

openrec/modeling/decoders/mdiff_decoder.py:240–247

(self, src, tgts, step_i=0)

        return res

    def forward_decoding(self, src, tgts, step_i=0):

        tgts = self.embedding(tgts)
        tgts = self.positional_encoding(tgts) + self.pos_embed
        for decoder_layer in self.decoder:
            tgts = decoder_layer(tgts, src, self_mask=None)

        return tgts

    def forward_reflect(self, src, pred_indexs, step_i=0):
        """Reflect decoding."""