| 161 | |
| 162 | |
| 163 | class MeanMapper(torch.nn.Module): |
| 164 | def __init__(self, preprocessing_dim): |
| 165 | super(MeanMapper, self).__init__() |
| 166 | self.preprocessing_dim = preprocessing_dim |
| 167 | |
| 168 | def forward(self, features): |
| 169 | features = features.reshape(len(features), 1, -1) |
| 170 | return F.adaptive_avg_pool1d(features, self.preprocessing_dim).squeeze(1) |
| 171 | |
| 172 | |
| 173 | class Aggregator(torch.nn.Module): |