libtorch c++输出接收方式

曙光_deeplove

已于 2024-03-13 09:38:09 修改

阅读量232

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习算法部署文章标签：部署

于 2024-03-12 08:43:50 首次发布

本文链接：https://blog.youkuaiyun.com/yangjinyi1314/article/details/136640691

28 篇文章 ¥49.90 ¥99.00

订阅专栏

本文介绍了在libtorch中如何处理C++端forward函数的输出，特别是当输出原本是PyTorch tracing网络的列表类型时，可以使用toList()方法进行接收。但需要注意的是，在高版本的libtorch中，此方法可能已不适用。

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

auto output = module.forward(inputs)

forward函数的输出，根据原始tracing的网络输出，对输出进行类别转换

比如我们原始tracing的网络输出的类型是列表类型，那么在c++端使用toList()去接，如下：

auto output = module.forward(inputs).toList();

另外备注：

（1）

toGenericListRef -> toListRef
isGenericList -> isList
toGenericList -> toList
toXListRef -> toXVector

（2）

toGenericList()

这个函数在高版本上无法进行列表接收了。