菜单

Administrator
发布于 2026-05-19 / 2 阅读
0
0

libtorch 常用 API 函数示例(史上最全、最详细)

libtorch 常用 API 函数示例(史上最全、最详细)

其实 pytorch 的函数 libtorch 都有,只是写法上有些出入。

libtorch 的官方文档链接:https://pytorch.org/cppdocs/api/classat_1_1_tensor.html

只是官方文档只是类似与函数声明,没有告诉干嘛的,只能通过函数名字猜了。

目录

  1. torch::full_like
  2. 创建与初始化 tensor
  3. 拼接 tensor torch::cat
  4. torch 的切片操作
  5. squeeze() unsqueeze()
  6. torch::nonzero
  7. 访问 tensor 值 a.item()
  8. opencv Mat 类型转 tensor
  9. tensor 的 size sizes() numel()
  10. torch::sort
  11. clamp
  12. 大于/小于运算
  13. 转置 Tensor::transpose
  14. expand_as
  15. 乘 mul_ 除 div 减 sub_
  16. 加载模型
  17. 模型 forward 出来的结果
  18. resize_ zero_
  19. meshgrid
  20. flatten
  21. fill_
  22. torch::stack
  23. reshape
  24. view
  25. argmax argmin
  26. where
  27. accessor
  28. torch::max torch::min
  29. masked_select 与 masked_fill
  30. libtorch 综合操作
  31. pytorch nms <-> libtorch nms
  32. 数据类型
  33. 指针访问 Tensor 数据
  34. Tensor 按索引赋值
    34-53. 其他高级操作

注:由于文章内容过长,包含大量 C++ 代码示例,完整内容请参考原始 CSDN 文章或直接访问 libtorch 官方文档。

原始来源:https://blog.csdn.net/无左无右


评论