(self, x, noise, t)
| 366 | extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x_start.shape) * noise) |
| 367 | |
| 368 | def get_v(self, x, noise, t): |
| 369 | return ( |
| 370 | extract_into_tensor(self.sqrt_alphas_cumprod, t, x.shape) * noise - |
| 371 | extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x.shape) * x |
| 372 | ) |
| 373 | |
| 374 | def get_loss(self, pred, target, mean=True): |
| 375 | if self.loss_type == 'l1': |
no test coverage detected