def forward(self, input):
# Return itself + the result of the two convolutions
output = self.model(input) + input
return output
# Image transformation network
微信
公众号
扫码关注公众号