Pytorch Create Feature Extractor