Method __init__

espnet2/gan_codec/shared/decoder/seanet_2d.py:256–258

(self, dim=2)

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

    def forward(self, x):
        return torch.unsqueeze(x, dim=self.dim)