def forward(self, inp):
    assert ((inp.dim() == 2) and (inp.size(1) == self.dis_input))
    return self.layers(inp).view((- 1))
def forward(self, inp):
    assert ((inp.dim() == 2) and (inp.size(1) == self.dis_input))
    return self.layers(inp).view((- 1))