raon_vision_encoder/utils.py

# Originally from OpenCLIP (https://github.com/mlfoundations/open_clip)

import collections.abc
from itertools import repeat


def _ntuple(n):
    def parse(x):
        if isinstance(x, collections.abc.Iterable):
            return x
        return tuple(repeat(x, n))

    return parse


to_2tuple = _ntuple(2)