DDPM අත්දැකීම් සඳහා උපයෝගිතා කාර්යයන්

10importtorch.utils.data

සිතියම්හැඩයට විශේෂාංග කිරීමට t සහ නැවත සකස් කිරීම සඳහා ගැටුම් එක්රැස් කරන්න

13defgather(consts:torch.Tensor,t:torch.Tensor):

15c=consts.gather(-1,t)16returnc.reshape(-1,1,1,1)