(self, src, src_mask, src_pos, map_emb)
| 72 | ) |
| 73 | |
| 74 | def encode(self, src, src_mask, src_pos, map_emb): |
| 75 | return self.encoder(self.src_embed(src), src_mask, src_pos, map_emb) |
| 76 | |
| 77 | def decode(self, memory, src_mask, tgt, tgt_mask, tgt_mask_agent, pos): |
| 78 |