src/transformers/models/mobilenet_v2/modeling_mobilenet_v2.py/MobileNetV2Stem/forward
class MobileNetV2Stem:    def forward(self, features: torch.Tensor) -> torch.Tensor:
        features = self.first_conv(features)
        if self.expand_1x1 is not None:
            features = self.expand_1x1(features)
        features = self.conv_3x3(features)
        features = self.reduce_1x1(features)
        return features