Method __init__

modules/audio2motion/flow_base.py:120–122

(self, *args)

class Permute(nn.Module):
    def __init__(self, *args):
        super(Permute, self).__init__()
        self.args = args

    def forward(self, x):
        return x.permute(self.args)