(self, x_start, t, noise=None)
| 354 | return_intermediates=return_intermediates) |
| 355 | |
| 356 | def q_sample(self, x_start, t, noise=None): |
| 357 | noise = default(noise, lambda: torch.randn_like(x_start)) |
| 358 | return (extract_into_tensor(self.sqrt_alphas_cumprod, t, x_start.shape) * x_start + |
| 359 | extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x_start.shape) * noise) |
| 360 | |
| 361 | def get_v(self, x, noise, t): |
| 362 | return ( |
no test coverage detected