def __call__(self, h):
    h = self.pixel_shuffle(h)
    h = self.c(h)
    if self.bn:
        h = self.b(h)
    if self.activation:
        h = self.activation(h)
    return h
def __call__(self, h):
    h = self.pixel_shuffle(h)
    h = self.c(h)
    if self.bn:
        h = self.b(h)
    if self.activation:
        h = self.activation(h)
    return h