def shortcut(l, n_in, n_out, stride):
    if (n_in != n_out):
        l = Conv2D('convshortcut', l, n_out, 1, stride=stride)
        return BatchNorm('bnshortcut', l)
    else:
        return l
def shortcut(l, n_in, n_out, stride):
    if (n_in != n_out):
        l = Conv2D('convshortcut', l, n_out, 1, stride=stride)
        return BatchNorm('bnshortcut', l)
    else:
        return l