def forward(self, x):
h = self.up1(x)
h = F.pad(h, (1, 1, 1, 1), mode='reflect')
h = self.b3(self.c2(h))
return F.relu(h)
微信
公众号
扫码关注公众号