【技术实现步骤摘要】
运算装置、芯片、反卷积运算方法、电子设备及存储介质
[0001]本公开涉人工智能
,尤其涉及一种运算装置、芯片、反卷积运算方法、电子设备及存储介质。
技术介绍
[0002]近些年来,人工智能技术取得了日新月异的发展,人工智能技术在数据的识别和处理等领域有了越来越广泛的应用。从根本上来讲,人工智能技术要建立在卷积、反卷积、池化、采样等基础运算上,因此这些基础运算的处理能力也决定了人工智能技术的运行能力。
[0003]以反卷积运算为例,相关技术中只能在GPU(graphics processing unit,图形处理器)、CPU(central processing unit,中央处理器)等支持矩阵运算和随机仿存的通用芯片上来实现反卷积运算,而无法在其他芯片上执行,尤其是不能在人工智能技术专用的加速芯片上执行,这不仅极大的降低了反卷积运算的效率,同时也限制了反卷积运算的硬件选配范围。
技术实现思路
[0004]本公开提供一种运算装置、芯片、反卷积运算方法、电子设备及存储介质,以解决相关技术中的缺陷。
[0005]根据本公开实施例的第一方面,提供一种运算装置,包括:
[0006]控制单元,用于获取反卷积运算的原始参数,根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数,并将所述目标参数发送至卷积运算单元,以及将所述等效卷积运算的结果确定为所述反卷积运算的结果;
[0007]卷积运算单元,用于根据所述目标参数进行所述等效卷积运算,并将所述等效卷积 ...
【技术保护点】
【技术特征摘要】
1.一种运算装置,其特征在于,包括:控制单元,用于获取反卷积运算的原始参数,根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数,并将所述目标参数发送至卷积运算单元,以及将所述等效卷积运算的结果确定为所述反卷积运算的结果;卷积运算单元,用于根据所述目标参数进行所述等效卷积运算,并将所述等效卷积运算的结果发送至所述控制单元。2.根据权利要求1所述的运算装置,其特征在于,所述原始参数包括下述至少一项:原始输入数据、原始卷积核、原始卷积步长、原始填充值以及输出数据的尺寸。3.根据权利要求2所述的运算装置,其特征在于,所述目标参数包括目标输入数据;所述控制单元用于根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数时,具体用于:根据所述原始输入数据的尺寸、所述原始卷积核的尺寸、所述原始卷积步长、所述原始填充值以及所述输出数据的尺寸,确定所述等效卷积运算的目标输入数据的尺寸,以及所述原始输入数据中的每个原始元素在所述目标输入数据中的位置;并根据所述原始输入数据中每个原始元素在所述目标输入数据中的位置,将所述原始输入数据中的每个原始元素添加到所述目标输入数据的对应位置,并在所述目标输入数据中的其他位置添加预设值,得到所述目标输入数据。4.根据权利要求3所述的运算装置,其特征在于,所述芯片还包括矢量运算单元和存储单元;所述控制单元用于根据所述原始输入数据中每个原始元素在所述目标输入数据中的位置,将所述原始输入数据中的每个原始元素添加到所述目标输入数据的对应位置,并在所述目标输入数据中的其他位置添加预设值,得到所述目标输入数据时,具体用于:根据所述目标输入数据的尺寸,在存储单元上获取对应尺寸的目标数据空间,并将所述目标数据空间中的每个位置上的目标元素初始化为所述预设值;并根据所述原始输入数据中的每个原始元素在所述目标输入数据中的位置,确定每个原始元素在所述目标数据空间中的位置;以及将所述原始输入数据的地址和每个原始元素在所述目标数据空间中的位置发送至所述矢量运算单元;所述存储单元用于:存储所述目标输入数据;所述矢量运算单元用于:根据所述原始输入数据的地址获取所述原始输入数据中的每个原始元素,并使用每个原始元素对所述目标数据空间中的对应位置上的目标元素进行更新。5.根据权利要求4所述的运算装置,其特征在于,所述控制单元用于将所述原始输入数据的地址和每个原始元素在所述目标数据空间中的位置发送至所述矢量运算单元时,具体用于:根据所述原始输入数据的地址、所述目标数据空间的地址、以及所述每个原始元素在所述目标数据空间中的位置,确定所述矢量运算单元的配置信息,并将所述配置信息发送至所述矢量运算单元,其中,所述矢量运算单元的配置信息包括下述至少一项:输入数据的首地...
【专利技术属性】
技术研发人员:勾志宏,胡英俊,徐宁仪,田志仲,
申请(专利权)人:上海阵量智能科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。