Encapsulating a prebuilt model in custom model

Question

I\'m trying to create a mobilenetV2forCifar10 model class by encapsulating a prebuilt MobilenetV2 model inside custom class and concatename the output tensor of the model to a p