Method forward

usr/diff/shallow_diffusion_tts.py:96–97

(self, x)

        self.g = nn.Parameter(torch.zeros(1))

    def forward(self, x):
        return self.fn(x) * self.g


# building block modules