Method __init__

usr/diff/shallow_diffusion_tts.py:44–46

(self, fn)

class Residual(nn.Module):
    def __init__(self, fn):
        super().__init__()
        self.fn = fn

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