def forward(self, input):
    output = self.relu(self.bn(self.conv(input)))
    output = torch.cat([output, self.pool(input)], 1)
    return output
def forward(self, input):
    output = self.relu(self.bn(self.conv(input)))
    output = torch.cat([output, self.pool(input)], 1)
    return output