libtorch 常用 API 函数示例(史上最全、最详细)
其实 pytorch 的函数 libtorch 都有,只是写法上有些出入。
libtorch 的官方文档链接:https://pytorch.org/cppdocs/api/classat_1_1_tensor.html
只是官方文档只是类似与函数声明,没有告诉干嘛的,只能通过函数名字猜了。
目录
- torch::full_like
- 创建与初始化 tensor
- 拼接 tensor torch::cat
- torch 的切片操作
- squeeze() unsqueeze()
- torch::nonzero
- 访问 tensor 值 a.item()
- opencv Mat 类型转 tensor
- tensor 的 size sizes() numel()
- torch::sort
- clamp
- 大于/小于运算
- 转置 Tensor::transpose
- expand_as
- 乘 mul_ 除 div 减 sub_
- 加载模型
- 模型 forward 出来的结果
- resize_ zero_
- meshgrid
- flatten
- fill_
- torch::stack
- reshape
- view
- argmax argmin
- where
- accessor
- torch::max torch::min
- masked_select 与 masked_fill
- libtorch 综合操作
- pytorch nms <-> libtorch nms
- 数据类型
- 指针访问 Tensor 数据
- Tensor 按索引赋值
34-53. 其他高级操作
注:由于文章内容过长,包含大量 C++ 代码示例,完整内容请参考原始 CSDN 文章或直接访问 libtorch 官方文档。