I have the checkpoint of a pre-trained model. I want to feed a tensor to this model as the input and get the output features (tensor). I don\'t want to restore the model using s