int main() {
torch::Tensor t = torch::ones({ 2, 3, 224, 224 });
ResNet resnet = resnet50();
t = resnet.forward(t);
std::cout << t.sizes() << t << std::endl;
}
This is my own writing a resnet50, by calling it, entering a tensor, and getting a result, you can print out the result. But for the network inside this link, I will not do the same operation.