(self, x)
| 21 | self.start_index = start_index |
| 22 | |
| 23 | def forward(self, x): |
| 24 | if self.start_index == 2: |
| 25 | readout = (x[:, 0] + x[:, 1]) / 2 |
| 26 | else: |
| 27 | readout = x[:, 0] |
| 28 | return x[:, self.start_index :] + readout.unsqueeze(1) |
| 29 | |
| 30 | |
| 31 | class ProjectReadout(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected