pytorch c++ 返回多值 - vector

代码

#include <torch/script.h>

#include <vector>

using namespace std;
using namespace at;

vector<Tensor> test_back(){
    torch::Tensor x = torch::randn({10,2});
    torch::Tensor y = torch::randn({10,2});
    return {x,y};
}

int main(int argc, const char* argv[])
{ 
    vector<Tensor> d;              
    d = test_back();
    cout << d[0].sizes() << endl;
    cout << d[1].sizes() << endl;

    cout<< "ok\n";
    return 1;
}

编译

make 
./bin/demo

输出

[10, 2]
[10, 2]
ok