(self, src, trg, src_mask, tgt_mask, src_padding_mask, tgt_padding_mask, memory_key_padding_mask)
| 64 | nn.init.xavier_uniform_(p) |
| 65 | |
| 66 | def forward(self, src, trg, src_mask, tgt_mask, src_padding_mask, tgt_padding_mask, memory_key_padding_mask): |
| 67 | |
| 68 | src_emb = self.pos_enc(self.src_embedding(src)) |
| 69 | tgt_emb = self.pos_enc(self.tgt_embedding(trg)) |
| 70 | |
| 71 | outs = self.transformer( |
| 72 | src_emb, |
| 73 | tgt_emb, |
| 74 | src_mask, |
| 75 | tgt_mask, |
| 76 | None, |
| 77 | src_padding_mask, |
| 78 | tgt_padding_mask, |
| 79 | memory_key_padding_mask |
| 80 | ) |
| 81 | |
| 82 | return self.ff(outs) |
| 83 | |
| 84 | def encode(self, src, src_mask): |
| 85 |
nothing calls this directly
no outgoing calls
no test coverage detected