def forward(self, x):
    input_1 = x
    output_3_1 = self.relu(self.conv_3_1(input_1))
    output_5_1 = self.relu(self.conv_5_1(input_1))
    input_2 = torch.cat([output_3_1, output_5_1], 1)
    output_3_2 = self.relu(self.conv_3_2(input_2))
    output_5_2 = self.relu(self.conv_5_2(input_2))
    input_3 = torch.cat([output_3_2, output_5_2], 1)
    output = self.confusion(input_3)
    output += x
    return output
def forward(self, x):
    input_1 = x
    output_3_1 = self.relu(self.conv_3_1(input_1))
    output_5_1 = self.relu(self.conv_5_1(input_1))
    input_2 = torch.cat([output_3_1, output_5_1], 1)
    output_3_2 = self.relu(self.conv_3_2(input_2))
    output_5_2 = self.relu(self.conv_5_2(input_2))
    input_3 = torch.cat([output_3_2, output_5_2], 1)
    output = self.confusion(input_3)
    output += x
    return output