(self, tensor, pos)
| 304 | self.activation = nn.ReLU() |
| 305 | |
| 306 | def with_pos_embed(self, tensor, pos): |
| 307 | return tensor + pos |
| 308 | |
| 309 | def forward(self, tgt, memory, mask= None, mask_ctx = None, pos= None, pos_ctx=None, gen_multi=False): |
| 310 | tgt2 = self.norm1(tgt) |