Sample xt from time-dependent density p_t; rng is required
(self, t, x0, x1)
| 119 | return alpha_t * x1 + sigma_t * x0 |
| 120 | |
| 121 | def compute_xt(self, t, x0, x1): |
| 122 | """Sample xt from time-dependent density p_t; rng is required""" |
| 123 | xt = self.compute_mu_t(t, x0, x1) |
| 124 | return xt |
| 125 | |
| 126 | def compute_ut(self, t, x0, x1, xt): |
| 127 | """Compute the vector field corresponding to p_t""" |