Method __init__

usr/diff/shallow_diffusion_tts.py:53–55

(self, dim)

class SinusoidalPosEmb(nn.Module):
    def __init__(self, dim):
        super().__init__()
        self.dim = dim

    def forward(self, x):
        device = x.device