(self, input_batch: NestedTensor)
| 175 | return self.linear(x) |
| 176 | |
| 177 | def score(self, input_batch: NestedTensor) -> Tensor: |
| 178 | y = self.predict(input_batch) |
| 179 | return y.sum(axis=-1) |
| 180 | |
| 181 | def predict_batch(self, input_batch: NestedTensor) -> NestedTensor: |
| 182 | x = input_batch["x"] |