libtorch : How to create a gpu tensor base on data_ptr?

前端 未结 0 1822
慢半拍i
慢半拍i 2020-12-23 20:36

create a gpu tensor base on data_ptr?

  int main{
        auto ten=torch::randn({3,10},torch::kCuda);
        auto p=ten.data_ptr();//I believe &         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题