用于在GPU上执行稀疏子流形卷积的方法和系统技术方案

技术编号:42651591 阅读:59 留言:0更新日期:2024-09-06 01:44
用于在GPU上执行稀疏子流形卷积的方法和系统。在图形处理单元上实施稀疏子流形卷积的方法包括:在图形处理单元接收密集格式的输入张量;在图形处理单元标识输入张量的活动位置;在图形处理单元基于已标识活动位置对输入张量执行索引展开运算,以生成一个输入矩阵,其中包含输入张量的每个活动窗口中的输入张量的元素;及在图形处理单元执行权重矩阵与输入矩阵之间的矩阵乘法,以生成输出矩阵,所述输出矩阵包括基于活动窗口的稀疏子流形卷积的输出张量的元素。所述方法还可能包括在图形处理单元基于活动窗口对输出矩阵执行索引折叠运算,以生成密集格式的输出张量。

【技术实现步骤摘要】

本申请涉及用于在图形处理单元(gpu)执行稀疏子流形卷积的方法和系统。


技术介绍

1、如本领域技术人员所已知,点云是在二维(2d)或三维(3d)空间中绘制的一组单独的数据点。例如,3d点云中的每个点可以表示在特定x、y和z位置处的测量值。点云可用于表示空间中的对象。点云可以通过传感器生成,例如但不限于lidar扫描仪或深度相机。如本领域技术人员所已知,lidar扫描仪使用脉冲激光器形式的光来测量距离。由于点云通常不具有每个可能坐标的点,因此点云被认为是稀疏数据集。

2、存在可运用点云的各种现实世界人工智能应用,例如增强╱虚拟现实(例如,内部场景的布局检测)和自主驾驶(例如,提取可驾驶区域)。因此,在点云上执行深度学习任务已经受到学术界和工业的极大关注,且人工神经网络(在本文中简称为神经网络)已被开发来处理点云,该网络在本文中可被称为点云神经网络。如本领域技术人员所已知,神经网络包括可用于机器学习应用的一个或多个互连层。具体地,神经网络可用于信号处理应用程序中,该信号处理应用程序包括但不限于图像处理应用程序和计算机视觉应用程序

3、本文档来自技高网...

【技术保护点】

1.一种在图形处理单元上实施稀疏子流形卷积的方法(500),该方法(500)包括:

2.根据权利要求1所述的方法(500),其中输入张量具有至少高度维度、宽度维度和通道维度,且输入张量的活动位置是其中输入张量的至少一个通道具有非零元素的高度和宽度位置。

3.根据权利要求2所述的方法(500),其中标识输入张量的活动位置包括:

4.根据权利要求1所述的方法(500),其中输入张量的活动窗口是输入张量的窗口,所述窗口用于生成稀疏子流形卷积的输出张量的元素,其中该窗口内的一个或多个预定位置是活动位置。

5.根据权利要求1至4中任一项所述的方法(5...

【技术特征摘要】

1.一种在图形处理单元上实施稀疏子流形卷积的方法(500),该方法(500)包括:

2.根据权利要求1所述的方法(500),其中输入张量具有至少高度维度、宽度维度和通道维度,且输入张量的活动位置是其中输入张量的至少一个通道具有非零元素的高度和宽度位置。

3.根据权利要求2所述的方法(500),其中标识输入张量的活动位置包括:

4.根据权利要求1所述的方法(500),其中输入张量的活动窗口是输入张量的窗口,所述窗口用于生成稀疏子流形卷积的输出张量的元素,其中该窗口内的一个或多个预定位置是活动位置。

5.根据权利要求1至4中任一项所述的方法(500),其中对输入张量执行索引展开运算包括从已标识的活动位置和稀疏子流形卷积的一个或多个参数中标识输入张量的活动窗口。

6.根据权利要求5所述的方法(500),其中从已标识的活动位置和稀疏子流形卷积的一个或多个参数中标识输入张量的活动窗口,就每个已标识的活动位置而言,包括:

7.根据权利要求5所述的方法(500),其中输入张量的活动窗口基于输入张量的被填充的版本,使得当稀疏子流形卷积在每个稀疏子流形卷积维度中的步幅为1时,每个活动位置都有活动窗口,以及从所识别的活动位置和稀疏子流形卷积的一个或多个参数中识别输入张量的活动窗口,包括识别对应于每个活动位置的活动窗口。

8.根据权利要求1至4中任一项所述的方法(500),其中对输入张量执行索引展开运算包括识别每个活动窗口中的输入张量元素。

9.根据权利要求8所述的方法(500),其中识别活动窗口中的输入张量元素包括实施一系列嵌套循环,该一系列嵌套循环包括活动窗口的每个维度的循环,该循环在所述维度上从窗口内的预定位置循环通过活动窗口的元素。

10.根据权利要求1至4中任一项所述的方法(500),其中对输入张量执行索引展开运算包括将各活动窗口的元素存储在输入矩阵中。

【专利技术属性】
技术研发人员:G·V·德米尔吉C·迪基奇G·M·史蒂文斯杨乐
申请(专利权)人:想象技术有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1