(self, model_output: torch.Tensor)
| 195 | self.model_outputs = [None] * self.order |
| 196 | |
| 197 | def update(self, model_output: torch.Tensor): |
| 198 | for i in range(self.order - 1): |
| 199 | self.model_outputs[i] = self.model_outputs[i + 1] |
| 200 | self.model_outputs[-1] = model_output |
| 201 | |
| 202 | def update_lower_order(self): |
| 203 | if self.lower_order_nums < self.order: |