(self, x, noise, t)
| 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 ( |
| 363 | extract_into_tensor(self.sqrt_alphas_cumprod, t, x.shape) * noise - |
| 364 | extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x.shape) * x |
| 365 | ) |
| 366 | |
| 367 | def get_loss(self, pred, target, mean=True): |
| 368 | if self.loss_type == 'l1': |
no test coverage detected