libtorch inference test

testpt.cpp

#include <torch/script.h>

int main()
{
    c10::InferenceMode guard(true);
    torch::NoGradGuard no_grad;

    torch::jit::Module mod2 = torch::jit::load("stylegan.pt");
    mod2.eval();

    at::Tensor t = torch::ones({1, 18, 512}, torch::kFloat32);

    std::vector<torch::jit::IValue> inputs;
    inputs.push_back(t);

    auto outputs = mod2.forward(inputs).toTensor();

    const int ndim = (int)outputs.dim();
    fprintf(stderr, "ndim = %d\n", ndim);

    return 0;
}