(self, layers, norm_layer=None, projection=None)
| 174 | Autoformer encoder |
| 175 | """ |
| 176 | def __init__(self, layers, norm_layer=None, projection=None): |
| 177 | super(Decoder, self).__init__() |
| 178 | self.layers = nn.ModuleList(layers) |
| 179 | self.norm = norm_layer |
| 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: |