Method forward

usr/diff/diffusion.py:48–49

(self, x, *args, **kwargs)

        self.fn = fn

    def forward(self, x, *args, **kwargs):
        return self.fn(x, *args, **kwargs) + x


class SinusoidalPosEmb(nn.Module):