(self, preprocessing_dim)
| 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) |