How to get torch::Tensor shape

前端 未结 0 1281
迷失自我
迷失自我 2021-02-19 06:04

If we << a torch::Tensor

torch::Tensor input_torch = torch::zeros({2, 3, 4});
cout << input_torch << endl;


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