(self, src, mask=None)
| 24 | self.norm = Norm(d_model) |
| 25 | |
| 26 | def forward(self, src, mask=None): |
| 27 | x = src |
| 28 | for i in range(self.N): |
| 29 | x = self.layers[i](x, mask) |
| 30 | return self.norm(x) |
| 31 | |
| 32 | class AE(nn.Module): |
| 33 |
nothing calls this directly
no outgoing calls
no test coverage detected