(self, src, src_mask)
| 82 | return self.ff(outs) |
| 83 | |
| 84 | def encode(self, src, src_mask): |
| 85 | |
| 86 | embed = self.src_embedding(src) |
| 87 | |
| 88 | pos_enc = self.pos_enc(embed) |
| 89 | |
| 90 | return self.transformer.encoder(pos_enc, src_mask) |
| 91 | |
| 92 | def decode(self, tgt, memory, tgt_mask): |
| 93 |