net.py 文件源码

def __call__(self, x, train=True):
        b, c, height, width = x.shape
        h = self.conv(F.unpooling_2d(x, 2, outsize=(height * 2, width * 2)))
        if self.activate:
            h = F.relu(self.bn(h, test=not train))
        return h