libtorch 常用api函数示例

GTX_AI

libtorch 常用api函数示例

1. torch::Tensor b = torch::argmax(output_1, 2).cpu();
   
2.     //    std::cout<<b<<std::endl;
   
3.     b.print();
   
4. 
   
5.     cv::Mat mask(T_height, T_width, CV_8UC1, (uchar*)b.data_ptr());
   
6.     imshow("mask",mask*255);
   
7.     waitKey(0);

复制代码

1. torch::Tensor a = torch::rand({2,3});
   
2.     torch::Tensor aa = a.clone();
   
3.     aa.masked_fill_(aa>0.5,-2);
   
4. 
   
5.     std::cout<<a<<std::endl;
   
6.     std::cout<<aa<<std::endl;

复制代码

1. 0.8803 0.2387 0.8577
   
2. 0.8166 0.0730 0.4682
   
3. [ Variable[CPUFloatType]{2,3} ]
   
4. -2.0000 0.2387 -2.0000
   
5. -2.0000 0.0730 0.4682
   
6. [ Variable[CPUFloatType]{2,3} ]

复制代码

参考：
【1】libtorch 常用api函数示例https://blog.csdn.net/yang332233/article/details/106199180