Yes sure!  Tou were right, there was a mistake writing the issue. Hope this helps you more @Nikronic !
Output before redimension torch.Size([10, 256, 13, 13])
RuntimeError                              Traceback (most recent call last)
 in ()
2 for param in modified_pretrained.parameters():
3    param.requires_grad = False
----> 4 (tensor_train, image_album) = feature_extr_cdist_train(modified_pretrained, load_train)
 in feature_extr_cdist_train(model, dataloader)
14         print(“Output before redimension”,outputs.size())
15         #print(“antes”, outputs.size())
—> 16         outputs = outputs.view(10, 432640)   #8028160
17
18         #print(outputs.size())
RuntimeError: shape ‘[10, 432640]’ is invalid for input of size 432640