pytorch c++ 返回多值 - vector

最新推荐文章于 2024-10-02 22:00:12 发布

.云哲.

最新推荐文章于 2024-10-02 22:00:12 发布

阅读量706

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： pytorch c++

本文链接：https://blog.youkuaiyun.com/luolinll1212/article/details/86156226

pytorch 专栏收录该内容

44 篇文章

订阅专栏

本文介绍了一个使用PyTorch库生成两个10x2随机张量的C++示例代码，并展示了如何通过编译和运行该代码来查看张量的大小。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码

#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;
}

编译