Method forward

selfpatch_vision_transformer.py:327–331

(self, x, loc=False)

        return self.pos_drop(x)

    def forward(self, x, loc=False):
        x = self.prepare_tokens(x)
        for blk in self.blocks:
            x = blk(x)
        return x

    def get_intermediate_layers(self, x, n=1):
        x = self.prepare_tokens(x)