src/transformers/models/bridgetower/modeling_bridgetower.py/BridgeTowerVisionTransformer/forward_pre
class BridgeTowerVisionTransformer:    def forward_pre(self, pixel_values: torch.Tensor):
        hidden_states = self.embeddings(pixel_values)
        hidden_states = self.ln_pre(hidden_states)
        # NLD -> LND
        hidden_states = hidden_states.permute(1, 0, 2)
        return hidden_states