torch::Tensor的文件存取

最新推荐文章于 2023-01-14 20:10:51 发布

IAmLinGe

最新推荐文章于 2023-01-14 20:10:51 发布

阅读量2.7k

点赞数

分类专栏： c++ 文章标签： torch::Tensor at::Tensor 存文件 pytorch c++

本文链接：https://blog.csdn.net/Betterc5/article/details/103206986

版权

c++ 专栏收录该内容

46 篇文章 2 订阅

订阅专栏

bool classA::saveCvecById(const std::string& user_id, at::Tensor cvec) {
#ifndef NDEBUG
    std::cerr << __FUNCTION__ << ": user_id: " << user_id << std::endl;
#endif
    std::string fileName = user_id + ".pt";
    torch::save(cvec, fileName);
    return true;
}

bool classA::readCvecById(const std::string& user_id, at::Tensor& cvec) {
#ifndef NDEBUG
    std::cerr << __FUNCTION__ << ": user_id: " << user_id << std::endl;
#endif
    std::string fileName = user_id + ".pt";
    torch::load(cvec, fileName);
    return true;
}