src/transformers/models/mobilevit/modeling_mobilevit.py/MobileViTDeepLabV3/forward
class MobileViTDeepLabV3:    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
        features = self.aspp(hidden_states[-1])
        features = self.dropout(features)
        features = self.classifier(features)
        return features