Compute the drift term of the SDE
(self, x, t)
| 165 | return self.d_log_mean_coeff(t) |
| 166 | |
| 167 | def compute_drift(self, x, t): |
| 168 | """Compute the drift term of the SDE""" |
| 169 | t = expand_t_like_x(t, x) |
| 170 | beta_t = self.sigma_min + (1 - t) * (self.sigma_max - self.sigma_min) |
| 171 | return -0.5 * beta_t * x, beta_t / 2 |
| 172 | |
| 173 | |
| 174 | class GVPCPlan(ICPlan): |
nothing calls this directly
no test coverage detected