Function flatten

byol_pytorch/byol_pytorch.py:17–18

(t)

    return def_val if val is None else val

def flatten(t):
    return t.reshape(t.shape[0], -1)

def singleton(cache_key):
    def inner_fn(fn):