| # 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) | |