技术标签: Pytorch c++ python LibTorch/模型部署 pytorch
目录
在PyTorch C++ API(libtorch)中对张量进行索引的方式与Python API的方式很相似。诸如None / ... / integer / boolean / slice / tensor的索引类型在C++ API里同样有效,这样就可以很方便的实现Python代码与C++代码的转换。主要的不同是将Python API里对张量的“[ ]”操作符转换成了以下形式:
torch::Tensor::index ( ) // 获取值
torch::Tensor::index_put_ ( ) // 设置值
有关官方文档请看这里。但是官方文档只简单地给了转换方式,没有具体例子有时还不太好理解。下面通过举例说明libtorch与pytorch中的向量索引/切片的方式,左边为Python方式,右边为C++方式:
1、tensor[Ellipsis, ...] --> tensor.index({Ellipsis, "..."})
import torch
a = torch.linspace(1,27,27).reshape(3, 3, 3)
print(a)
c = a[..., 2]
print(c)
#===================运行结果===============#
tensor([[[ 1., 2., 3.],
[ 4., 5., 6.],
[ 7., 8., 9.]],
[[10., 11., 12.],
[13., 14., 15.],
[16., 17., 18.]],
[[19., 20., 21.],
[22., 23., 24.],
[25., 26., 27.]]])
tensor([[ 3., 6., 9.],
[12., 15., 18.],
[21., 24., 27.]])
#include "iostream"
#include "torch/script.h"
int main()
{
torch::Tensor a = torch::linspace(1, 27, 27).reshape({3, 3, 3});
std::cout << a << std::endl;
at::Tensor b = a.index({"...", 2});
std::cout << b << std::endl;
return 0;
}
/****************输出结果******************/
(1,.,.) =
1 2 3
4 5 6
7 8 9
(2,.,.) =
10 11 12
13 14 15
16 17 18
(3,.,.) =
19 20 21
22 23 24
25 26 27
[ CPUFloatType{3,3,3} ]
3 6 9
12 15 18
21 24 27
[ CPUFloatType{3,3} ]
2、tensor[1, 2] --> tensor.index({1, 2})
import torch
a = torch.linspace(1,27,27).reshape(3, 3, 3)
print(a)
c = a[1, 2]
print(c)
#===================运行结果=================#
tensor([[[ 1., 2., 3.],
[ 4., 5., 6.],
[ 7., 8., 9.]],
[[10., 11., 12.],
[13., 14., 15.],
[16., 17., 18.]],
[[19., 20., 21.],
[22., 23., 24.],
[25., 26., 27.]]])
tensor([16., 17., 18.])
#include "iostream"
#include "torch/script.h"
int main()
{
torch::Tensor a = torch::linspace(1, 27, 27).reshape({3, 3, 3});
std::cout << a << std::endl;
at::Tensor b = a.index({1, 2});
std::cout << b << std::endl;
return 0;
}
/*****************运行结果***************/
(1,.,.) =
1 2 3
4 5 6
7 8 9
(2,.,.) =
10 11 12
13 14 15
16 17 18
(3,.,.) =
19 20 21
22 23 24
25 26 27
[ CPUFloatType{3,3,3} ]
16
17
18
[ CPUFloatType{3} ]
3、tensor[1::2] --> tensor.index({Slice(1, None, 2)})
import torch
a = torch.linspace(1, 6, 6)
print(a)
c = a[1::2]
print(c)
#==================运行结果==================#
tensor([1., 2., 3., 4., 5., 6.])
tensor([2., 4., 6.])
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::linspace(1, 6, 6);
std::cout << a << std::endl;
at::Tensor b = a.index({Slice(1, None, 2)});
std::cout << b << std::endl;
return 0;
}
/*******************运行结果*********************/
1
2
3
4
5
6
[ CPUFloatType{6} ]
2
4
6
[ CPUFloatType{3} ]
3.33、tensor[..., 1:] --> tensor.index({"...", Slice(1)})
import torch
a = torch.linspace(1,27,27).reshape(3, 3, 3)
b = a[..., 1:]
print(b)
#===============运行结果===================#
tensor([[[ 2., 3.],
[ 5., 6.],
[ 8., 9.]],
[[11., 12.],
[14., 15.],
[17., 18.]],
[[20., 21.],
[23., 24.],
[26., 27.]]])
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::linspace(1, 27, 27).reshape({3, 3, 3});
torch::Tensor b = a.index({"...", Slice(1)});
std::cout << b << std::endl;
return 0;
}
/******************运行结果**********************/
(1,.,.) =
2 3
5 6
8 9
(2,.,.) =
11 12
14 15
17 18
(3,.,.) =
20 21
23 24
26 27
[ CPUFloatType{3,3,2} ]
3.66、tensor[..., :2] --> tensor.index({"...", Slice({None, 2})})
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
float a[2][3][3] = {
{
{1,2,2},{3,4,4},{5,6,6}},{
{1,2,2},{3,4,4},{5,6,6}}};
at::Tensor b = at::from_blob(a, {2, 3, 3}, at::kFloat);
at::Tensor c = b.index({"...", Slice({None, 2})}); // 留下最后一维的前两列,相当于Python中的b[..., :2]
auto d = c.sizes();
std::cout << b << std::endl;
std::cout << c << std::endl;
std::cout << d << std::endl;
return 0;
}
/********************运行结果********************/
(1,.,.) =
1 2 2
3 4 4
5 6 6
(2,.,.) =
1 2 2
3 4 4
5 6 6
[ CPUFloatType{2,3,3} ]
(1,.,.) =
1 2
3 4
5 6
(2,.,.) =
1 2
3 4
5 6
[ CPUFloatType{2,3,2} ]
[2, 3, 2]
4、tensor[torch.tensor([1, 2])] --> tensor.index({torch::tensor({1, 2})})
import torch
a = torch.linspace(1,4,4)
b = torch.tensor([0, 1, 3, 2])
c = a[b]
print(a)
print(c)
#===============运行结果===============#
tensor([1., 2., 3., 4.])
tensor([1., 2., 4., 3.])
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::linspace(1, 4, 4);
torch::Tensor b = torch::tensor({0, 1, 3, 2});
torch::Tensor c = a.index({b});
std::cout << a << std::endl;
std::cout << b << std::endl;
return 0;
}
/*******************运行结果********************/
1
2
3
4
[ CPUFloatType{4} ]
0
1
3
2
[ CPULongType{4} ]
1、tensor[1, 2] = 1 --> tensor.index_put_({1, 2}, 1)
import torch
a = torch.linspace(1,4,4).reshape(2, 2)
print(a)
a[1, 1] = 100
print(a)
#==================运行结果=====================#
tensor([[1., 2.],
[3., 4.]])
tensor([[ 1., 2.],
[ 3., 100.]])
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::linspace(1, 4, 4).reshape({2, 2});
std::cout << a << std::endl;
a.index_put_({1, 1}, 100);
std::cout << a << std::endl;
return 0;
}
/***************运行结果****************/
1 2
3 4
[ CPUFloatType{2,2} ]
1 2
3 100
[ CPUFloatType{2,2} ]
2、tensor[Ellipsis, ...] = 1 --> tensor.index_put_({Ellipsis, "..."}, 1)
import torch
a = torch.linspace(1,4,4).reshape(2, 2)
print(a)
a[..., 1] = 100
print(a)
#====================运行结果=====================#
tensor([[1., 2.],
[3., 4.]])
tensor([[ 1., 100.],
[ 3., 100.]])
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::linspace(1, 4, 4).reshape({2, 2});
std::cout << a << std::endl;
a.index_put_({"...", 1}, 100);
std::cout << a << std::endl;
return 0;
}
/***************运行结果****************/
1 2
3 4
[ CPUFloatType{2,2} ]
1 100
3 100
[ CPUFloatType{2,2} ]
3、tensor[torch.tensor([1, 2])] = 1 --> tensor.index_put_({torch::tensor({1, 2})}, 1)
import torch
a = torch.linspace(1,4,4)
b = torch.tensor([0, 2])
print(a)
a[b] = 100
print(a)
#===============运行结果==================#
tensor([1., 2., 3., 4.])
tensor([100., 2., 100., 4.])
#include "iostream"
#include "torch/script.h"
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::linspace(1, 4, 4);
torch::Tensor b = torch::tensor({0, 2});
std::cout << a << std::endl;
a.index_put_({b}, 100);
std::cout << a << std::endl;
return 0;
}
/*****************运行结果*****************/
1
2
3
4
[ CPUFloatType{4} ]
100
2
100
4
[ CPUFloatType{4} ]
按理说这部分也算索引的一部分,为什么要拿出来单独说呐?因为他比较特殊。首先看下例子吧。
在PyTorch中,我们使用Python的切片方式可以方便地通过掩码向量或矩阵来获取想要的tensor区域,比如以下操作:
import torch
a = torch.randint(0, 9,(4,4)) # 创建shape为4*4,值为[0,9]的随机整数的tensor
b = torch.tensor([0, 1, 0, 1]).bool() # 创建bool值向量,最终的结果是对应行向量的取舍
c = torch.tensor([0, 1, 0, 1]) # 注意这里不是bool值,最终的结果只是按行索引
a1 = a[b] # 掩码操作
a2 = a[c] # 索引操作
print('a:', a)
print('b:', b)
print('c:', c)
print('a1:', a1)
print('a2:', a2)
#=================运行结果===================#
a: tensor([[8, 0, 4, 8],
[5, 7, 7, 3],
[0, 5, 8, 1],
[3, 3, 2, 1]])
b: tensor([False, True, False, True])
c: tensor([0, 1, 0, 1])
a1: tensor([[5, 7, 7, 3],
[3, 3, 2, 1]])
a2: tensor([[8, 0, 4, 8],
[5, 7, 7, 3],
[8, 0, 4, 8],
[5, 7, 7, 3]])
在PyThon的C++API即LibTorch中,也有类似的操作,举例如下:
#include "iostream"
#include "ATen/Parallel.h"
#include "torch/script.h"
#include <c10/core/ScalarType.h>
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::randint(0, 9, {4,4}); // 初始化
torch::Tensor b = at::tensor({0,1,0,1}).toType(at::kBool); // bool型掩码向量,注意与非bool型的结果区别
torch::Tensor c = at::tensor({0,1,0,1}).toType(at::kLong); // 非bool型按行索引(这里的索引用tensor必须是long, byte or bool类型的)
torch::Tensor a1 = a.index({b}); // 掩码操作
torch::Tensor a2 = a.index({c}); // 按行索引
std::cout << a << std::endl;
std::cout << b << std::endl;
std::cout << c << std::endl;
std::cout << a1 << std::endl;
std::cout << a2 << std::endl;
return 0;
}
//==================运行结果====================//
7 5 8 0
1 0 1 2
0 6 6 3
2 1 5 3
[ CPUFloatType{4,4} ]
0
1
0
1
[ CPUBoolType{4} ]
0
1
0
1
[ CPULongType{4} ]
1 0 1 2
2 1 5 3
[ CPUFloatType{2,4} ]
7 5 8 0
1 0 1 2
7 5 8 0
1 0 1 2
[ CPUFloatType{4,4} ]
所以一定要注意0和1在bool型与非bool型情况下对索引结果的影响!
另外,LibTorch还有一个masked_select()方法,该方法是按照对应位置上的掩码值(bool值)来抽取值,先看定义:
inline Tensor Tensor::masked_select(const Tensor & mask)
其中mask必须是与要操作的tensor同shape或拥有相同数量元素的tensor,不满足条件就会自动复制补全,直至与目标tensor具有相同数量的元素。
举例如下:
#include "iostream"
#include "torch/script.h"
#include <c10/core/ScalarType.h>
using namespace torch::indexing;
int main()
{
torch::Tensor a = torch::randint(0, 9, {4,4}); // 初始化
torch::Tensor b = a > 5; // 获取满足条件的bool矩阵
torch::Tensor c = a.masked_select({b}); // 按照bool矩阵来取舍对应的值
std::cout << a << std::endl;
std::cout << b << std::endl;
std::cout << c << std::endl;
return 0;
}
//====================运行结果=======================//
5 2 8 8
1 0 3 0
2 2 4 4
8 4 7 5
[ CPUFloatType{4,4} ]
0 0 1 1
0 0 0 0
0 0 0 0
1 0 1 0
[ CPUBoolType{4,4} ]
8
8
8
7
[ CPUFloatType{4} ]
注:这里当时测试用的具体libtorch版本记不清了,不同版本间的操作可能会有差异
文章浏览阅读3.4k次,点赞8次,收藏42次。一、什么是内部类?or 内部类的概念内部类是定义在另一个类中的类;下面类TestB是类TestA的内部类。即内部类对象引用了实例化该内部对象的外围类对象。public class TestA{ class TestB {}}二、 为什么需要内部类?or 内部类有什么作用?1、 内部类方法可以访问该类定义所在的作用域中的数据,包括私有数据。2、内部类可以对同一个包中的其他类隐藏起来。3、 当想要定义一个回调函数且不想编写大量代码时,使用匿名内部类比较便捷。三、 内部类的分类成员内部_成员内部类和局部内部类的区别
文章浏览阅读118次。分布式系统要求拆分分布式思想的实质搭配要求分布式系统要求按照某些特定的规则将项目进行拆分。如果将一个项目的所有模板功能都写到一起,当某个模块出现问题时将直接导致整个服务器出现问题。拆分按照业务拆分为不同的服务器,有效的降低系统架构的耦合性在业务拆分的基础上可按照代码层级进行拆分(view、controller、service、pojo)分布式思想的实质分布式思想的实质是为了系统的..._分布式系统运维工具
文章浏览阅读174次。1.数据源准备2.数据处理step1:数据表处理应用函数:①VLOOKUP函数; ② CONCATENATE函数终表:step2:数据透视表统计分析(1) 透视表汇总不同渠道用户数, 金额(2)透视表汇总不同日期购买用户数,金额(3)透视表汇总不同用户购买订单数,金额step3:讲第二步结果可视化, 比如, 柱形图(1)不同渠道用户数, 金额(2)不同日期..._exce l趋势分析数据量
文章浏览阅读3.3k次。堡垒机可以为企业实现服务器、网络设备、数据库、安全设备等的集中管控和安全可靠运行,帮助IT运维人员提高工作效率。通俗来说,就是用来控制哪些人可以登录哪些资产(事先防范和事中控制),以及录像记录登录资产后做了什么事情(事后溯源)。由于堡垒机内部保存着企业所有的设备资产和权限关系,是企业内部信息安全的重要一环。但目前出现的以下问题产生了很大安全隐患:密码设置过于简单,容易被暴力破解;为方便记忆,设置统一的密码,一旦单点被破,极易引发全面危机。在单一的静态密码验证机制下,登录密码是堡垒机安全的唯一_horizon宁盾双因素配置
文章浏览阅读7.7k次,点赞4次,收藏16次。Chrome作为一款挺不错的浏览器,其有着诸多的优良特性,并且支持跨平台。其支持(Windows、Linux、Mac OS X、BSD、Android),在绝大多数情况下,其的安装都很简单,但有时会由于网络原因,无法安装,所以在这里总结下Chrome的安装。Windows下的安装:在线安装:离线安装:Linux下的安装:在线安装:离线安装:..._chrome linux debian离线安装依赖
文章浏览阅读153次。中国发达城市榜单每天都在刷新,但无非是北上广轮流坐庄。北京拥有最顶尖的文化资源,上海是“摩登”的国际化大都市,广州是活力四射的千年商都。GDP和发展潜力是衡量城市的数字指...
文章浏览阅读3.3k次。前言spark在java使用比较少,多是scala的用法,我这里介绍一下我在项目中使用的代码配置详细算法的使用请点击我主页列表查看版本jar版本说明spark3.0.1scala2.12这个版本注意和spark版本对应,只是为了引jar包springboot版本2.3.2.RELEASEmaven<!-- spark --> <dependency> <gro_使用java调用spark注册进去的程序
文章浏览阅读4.8k次。汽车零部件开发工具巨头V公司全套bootloader中UDS协议栈源代码,自己完成底层外设驱动开发后,集成即可使用,代码精简高效,大厂出品有量产保证。:139800617636213023darcy169_uds协议栈 源代码
文章浏览阅读4.6k次,点赞20次,收藏148次。AUTOSAR基础篇之OS(下)前言首先,请问大家几个小小的问题,你清楚:你知道多核OS在什么场景下使用吗?多核系统OS又是如何协同启动或者关闭的呢?AUTOSAR OS存在哪些功能安全等方面的要求呢?多核OS之间的启动关闭与单核相比又存在哪些异同呢?。。。。。。今天,我们来一起探索并回答这些问题。为了便于大家理解,以下是本文的主题大纲:[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-JCXrdI0k-1636287756923)(https://gite_autosar 定义了 5 种多核支持类型
文章浏览阅读2.2k次,点赞6次,收藏14次。原因:自己写的头文件没有被加入到方案的包含目录中去,无法被检索到,也就无法打开。将自己写的头文件都放入header files。然后在VS界面上,右键方案名,点击属性。将自己头文件夹的目录添加进去。_vs2013打不开自己定义的头文件
文章浏览阅读3.3w次,点赞80次,收藏342次。此时,可以将系统中所有用户的 Session 数据全部保存到 Redis 中,用户在提交新的请求后,系统先从Redis 中查找相应的Session 数据,如果存在,则再进行相关操作,否则跳转到登录页面。此时,可以将系统中所有用户的 Session 数据全部保存到 Redis 中,用户在提交新的请求后,系统先从Redis 中查找相应的Session 数据,如果存在,则再进行相关操作,否则跳转到登录页面。当数据量很大时,count 的数量的指定可能会不起作用,Redis 会自动调整每次的遍历数目。_redis命令
文章浏览阅读449次,点赞3次,收藏3次。URP的设计目标是在保持高性能的同时,提供更多的渲染功能和自定义选项。与普通项目相比,会多出Presets文件夹,里面包含着一些设置,包括本色,声音,法线,贴图等设置。全局只有主光源和附加光源,主光源只支持平行光,附加光源数量有限制,主光源和附加光源在一次Pass中可以一起着色。URP:全局只有主光源和附加光源,主光源只支持平行光,附加光源数量有限制,一次Pass可以计算多个光源。可编程渲染管线:渲染策略是可以供程序员定制的,可以定制的有:光照计算和光源,深度测试,摄像机光照烘焙,后期处理策略等等。_urp渲染管线