(self, tgt, memory, tgt_mask)
| 90 | return self.transformer.encoder(pos_enc, src_mask) |
| 91 | |
| 92 | def decode(self, tgt, memory, tgt_mask): |
| 93 | |
| 94 | embed = self.tgt_embedding(tgt) |
| 95 | |
| 96 | pos_enc = self.pos_enc(embed) |
| 97 | |
| 98 | return self.transformer.decoder(pos_enc, memory, tgt_mask) |