def prepare_tokens(self, x):
    (B, nc, w, h) = x.shape
    x = self.patch_embed(x)
    cls_tokens = self.cls_token.expand(B, (- 1), (- 1))
    x = torch.cat((cls_tokens, x), dim=1)
    x = (x + self.interpolate_pos_encoding(x, w, h))
    return self.pos_drop(x)
def prepare_tokens(self, x):
    (B, nc, w, h) = x.shape
    x = self.patch_embed(x)
    cls_tokens = self.cls_token.expand(B, (- 1), (- 1))
    x = torch.cat((cls_tokens, x), dim=1)
    x = (x + self.interpolate_pos_encoding(x, w, h))
    return self.pos_drop(x)