(self)
| 58 | return (self.mu, self.sigma / np.sqrt(2 * self.theta)) |
| 59 | |
| 60 | def reset(self): |
| 61 | self.x = self.randn(*self.shape) * self.sigma / np.sqrt(2 * self.theta) + self.mu |
| 62 | |
| 63 | def reset_part(self, mask: np.ndarray): |
| 64 | # mask: bool array with same shape as self.x |