![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
LibTorch
LibTorch
小鹏AI
深度学习算法工程师,主攻深度学习模型优化部署,辅攻图形图像算法调优。
中级软件设计师、英伟达Jeston Nano专家认证。
展开
-
C++中如何将Tensor类型转换为基本类型呢?
int a = torch::Tensor(123).item<int>();原创 2021-12-10 17:46:41 · 1539 阅读 · 1 评论 -
LibTorch 使用torch::from_blob出现的问题
LibTorch 使用torch::from_blob出现的问题原创 2022-11-10 15:09:05 · 1295 阅读 · 0 评论 -
torch::Tensor 支持的类型有哪些?
/// Fixed width dtypes.constexpr auto kUInt8 = at::kByte;constexpr auto kInt8 = at::kChar;constexpr auto kInt16 = at::kShort;constexpr auto kInt32 = at::kInt;constexpr auto kInt64 = at::kLong;constexpr auto kFloat16 = at::kHalf;constexpr auto kFloat原创 2021-12-10 14:58:23 · 933 阅读 · 0 评论 -
TorchScriptModule 转换并将模型保存
TorchScriptModule 转换并将模型保存。原创 2022-11-10 14:56:59 · 300 阅读 · 0 评论