def add_noise

函数功能

在原始样本中添加噪声。

函数原型

def add_noise(self, original_samples: torch.FloatTensor, noise: torch.FloatTensor, timesteps: torch.IntTensor) -> torch.FloatTensor:

参数说明

返回值说明

返回添加噪声后的结果。