Function make_2tuple

vtp/models/layers/embeddings.py:9–15

Convert input to 2-tuple.

(x)

def make_2tuple(x):
    """Convert input to 2-tuple."""
    if isinstance(x, tuple):
        assert len(x) == 2
        return x
    assert isinstance(x, int)
    return (x, x)


class PatchEmbed(nn.Module):