models.py 文件源码

def differentiable_backward(self, g):
        g = self.c_last.differentiable_backward(g)
        _b, _ch, _w, _h = self.last_shape
        g = F.reshape(g, (_b, _ch, _w, _h))
        for i in reversed(range(self.down_layers-1)):
            g = getattr(self, 'c'+str(i)).differentiable_backward(g)
        g = self.c_first.differentiable_backward(g)
        return g