def shortcut(self, x):
    if self.learnable_sc:
        x = self.c_sc(x)
        if self.sample:
            return self.sample(x)
        else:
            return x
    else:
        return x
def shortcut(self, x):
    if self.learnable_sc:
        x = self.c_sc(x)
        if self.sample:
            return self.sample(x)
        else:
            return x
    else:
        return x