4ffda89c42361af48c15721b2739b36c52f053d6,layers/tacotron2.py,LinearBN,forward,#LinearBN#Any#,45

Before Change

    def forward(self, x):
        out = self.linear_layer(x)
        out = out.transpose(0, 1).transpose(1, 2)
        out = self.bn(out)
        out = out.transpose(1, 2).transpose(0, 1)
        return out


class Prenet(nn.Module):

After Change

        if len(out.shape)==3:
            out = out.permute(1, 2, 0)
        out = self.bn(out)
        if len(out.shape) == 3:
            out = out.permute(2, 0, 1)
        return out


class Prenet(nn.Module):

Instances