Function cast_tuple

sat/sgm/modules/autoencoding/magvit2_pytorch.py:89–90

(t, length=1)

def cast_tuple(t, length=1):
    return t if isinstance(t, tuple) else ((t,) * length)


# tensor helpers