pytorch c++ 环境搭建,点击
#include <torch/script.h>
#include <ATen/ATen.h>
#include <iostream>
#include <memory>
using namespace std;
using namespace at;
int main(int argc, const char* argv[])
{
torch::Tensor a = torch::randn({2,3});
std::cout<< a << std::endl;
std::cout<< a.sizes() << std::endl;
auto b = at::transpose(a,1,0);
std::cout<< b << std::endl;
std::cout<< b.sizes() << std::endl;
std::cout<< "--------------------" << std::endl;
torch::Tensor x = torch::randn({4,6});
std::cout<< x.sizes() << std::endl;
auto y = at::_th_view(x,{2,12});
std::cout<< y.sizes() << std::endl;
std::cout<< "--------------------" << std::endl;
torch::Tensor m = torch::randn({4,6});
std::cout<< m.sizes() << std::endl;
auto n = at::_th_view(m,{2,3,4});
std::cout<< n.sizes() << std::endl;
std::cout<< "--------------------" << std::endl;
std::cout<< "ok\n";
return 1;
}
编译并执行
make clean
make
./bin/demo
输出结果
rose@rose-machine:~/pytorch_c++/test$ ./bin/demo
0.0786 -0.8607 0.3106
-0.5987 -0.3663 0.4673
[ Variable[CPUFloatType]{2,3} ]
[2, 3]
0.0786 -0.5987
-0.8607 -0.3663
0.3106 0.4673
[ Variable[CPUFloatType]{3,2} ]
[3, 2]
--------------------
[4, 6]
[2, 12]
--------------------
[4, 6]
[2, 3, 4]
--------------------
ok