(self, x, cross, x_mask=None, cross_mask=None, trend=None)
| 180 | self.projection = projection |
| 181 | |
| 182 | def forward(self, x, cross, x_mask=None, cross_mask=None, trend=None): |
| 183 | for layer in self.layers: |
| 184 | x, residual_trend = layer(x, cross, x_mask=x_mask, cross_mask=cross_mask) |
| 185 | trend = trend + residual_trend |
| 186 | |
| 187 | if self.norm is not None: |
| 188 | x = self.norm(x) |
| 189 | |
| 190 | if self.projection is not None: |
| 191 | x = self.projection(x) |
| 192 | return x, trend |
nothing calls this directly
no outgoing calls
no test coverage detected