src/transformers/models/dinat/modeling_dinat.py/DinatDownsampler/forward
class DinatDownsampler:    def forward(self, input_feature: torch.Tensor) -> torch.Tensor:
        input_feature = self.reduction(input_feature.permute(0, 3, 1, 2)).permute(0, 2, 3, 1)
        input_feature = self.norm(input_feature)
        return input_feature