运算装置、芯片、反卷积运算方法、电子设备及存储介质制造方法及图纸

技术编号:34521409 阅读:9 留言:0更新日期:2022-08-13 21:10
本公开涉及一种运算装置、芯片、反卷积运算方法、电子设备及存储介质,所述装置包括:控制单元,用于获取反卷积运算的原始参数,根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数,并将所述目标参数发送至卷积运算单元,以及将所述等效卷积运算的结果确定为所述反卷积运算的结果;卷积运算单元,用于根据所述目标参数进行所述等效卷积运算,并将所述等效卷积运算的结果发送至所述控制单元。至所述控制单元。至所述控制单元。

【技术实现步骤摘要】
运算装置、芯片、反卷积运算方法、电子设备及存储介质


[0001]本公开涉人工智能
,尤其涉及一种运算装置、芯片、反卷积运算方法、电子设备及存储介质。

技术介绍

[0002]近些年来,人工智能技术取得了日新月异的发展,人工智能技术在数据的识别和处理等领域有了越来越广泛的应用。从根本上来讲,人工智能技术要建立在卷积、反卷积、池化、采样等基础运算上,因此这些基础运算的处理能力也决定了人工智能技术的运行能力。
[0003]以反卷积运算为例,相关技术中只能在GPU(graphics processing unit,图形处理器)、CPU(central processing unit,中央处理器)等支持矩阵运算和随机仿存的通用芯片上来实现反卷积运算,而无法在其他芯片上执行,尤其是不能在人工智能技术专用的加速芯片上执行,这不仅极大的降低了反卷积运算的效率,同时也限制了反卷积运算的硬件选配范围。

技术实现思路

[0004]本公开提供一种运算装置、芯片、反卷积运算方法、电子设备及存储介质,以解决相关技术中的缺陷。
[0005]根据本公开实施例的第一方面,提供一种运算装置,包括:
[0006]控制单元,用于获取反卷积运算的原始参数,根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数,并将所述目标参数发送至卷积运算单元,以及将所述等效卷积运算的结果确定为所述反卷积运算的结果;
[0007]卷积运算单元,用于根据所述目标参数进行所述等效卷积运算,并将所述等效卷积运算的结果发送至所述控制单元。
[0008]在一个实施例中,所述原始参数包括下述至少一项:
[0009]原始输入数据、原始卷积核、原始卷积步长、原始填充值以及输出数据的尺寸。
[0010]在一个实施例中,所述目标参数包括目标输入数据;
[0011]所述控制单元用于根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数时,具体用于:根据所述原始输入数据的尺寸、所述原始卷积核的尺寸、所述原始卷积步长、所述原始填充值以及所述输出数据的尺寸,确定所述等效卷积运算的目标输入数据的尺寸,以及所述原始输入数据中的每个原始元素在所述目标输入数据中的位置;并根据所述原始输入数据中每个原始元素在所述目标输入数据中的位置,将所述原始输入数据中的每个原始元素添加到所述目标输入数据的对应位置,并在所述目标输入数据中的其他位置添加预设值,得到所述目标输入数据。
[0012]在一个实施例中,所述芯片还包括矢量运算单元和存储单元;
[0013]所述控制单元用于根据所述原始输入数据中每个原始元素在所述目标输入数据
中的位置,将所述原始输入数据中的每个原始元素添加到所述目标输入数据的对应位置,并在所述目标输入数据中的其他位置添加预设值,得到所述目标输入数据时,具体用于:根据所述目标输入数据的尺寸,在存储单元上获取对应尺寸的目标数据空间,并将所述目标数据空间中的每个位置上的目标元素初始化为所述预设值;并根据所述原始输入数据中的每个原始元素在所述目标输入数据中的位置,确定每个原始元素在所述目标数据空间中的位置;以及将所述原始输入数据的地址和每个原始元素在所述目标数据空间中的位置发送至所述矢量运算单元;
[0014]所述存储单元用于:存储所述目标输入数据;
[0015]所述矢量运算单元用于:根据所述原始输入数据的地址获取所述原始输入数据中的每个原始元素,并使用每个原始元素对所述目标数据空间中的对应位置上的目标元素进行更新。
[0016]在一个实施例中,所述控制单元用于将所述原始输入数据的地址和每个原始元素在所述目标数据空间中的位置发送至所述矢量运算单元时,具体用于:根据所述原始输入数据的地址、所述目标数据空间的地址、以及所述每个原始元素在所述目标数据空间中的位置,确定所述矢量运算单元的配置信息,并将所述配置信息发送至所述矢量运算单元,其中,所述矢量运算单元的配置信息包括下述至少一项:输入数据的首地址、运算步长和每个数据维度的尺寸值,以及输出数据的首地址、运算步长、每个数据维度的尺寸值和每个数据维度的循环值;
[0017]所述矢量运算单元用于根据所述原始输入数据的地址获取所述原始输入数据中的每个原始元素,并使用每个原始元素对所述目标数据空间中的对应位置上的目标元素进行更新时,具体用于:根据所述配置信息,将每个原始元素在所述目标数据空间中的位置上的目标元素更新为对应的原始元素与所述预设值的和。
[0018]在一个实施例中,所述目标参数包括目标卷积核;
[0019]所述控制单元用于根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数时,具体用于:将所述原始卷积核中的数据进行中心对称旋转,得到所述等效卷积运算的目标卷积核。
[0020]在一个实施例中,所述原始卷积核包括多个通道;
[0021]所述控制单元用于根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数时,具体用于:将所述原始卷积核中每个通道的数据分别进行中心对称旋转,得到所述目标卷积核的每个通道的数据。
[0022]在一个实施例中,所述目标参数包括目标卷积步长和/或目标填充值;
[0023]所述控制单元用于根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数时,具体用于:将所述等效卷积运算的目标卷积步长确定为预设步长;和/或,将所述等效卷积运算的目标填充值确定为预设填充值。
[0024]在一个实施例中,所述控制单元用于将所述目标参数发送至卷积运算单元时,具体用于:将所述目标输入数据的首地址和每个数据维度的尺寸值、所述目标卷积核的首地址和每个数据维度的尺寸值、所述目标卷积步长、所述目标填充值配置至所述卷积运算单元对应的寄存器中;
[0025]所述矢量运算单元用于根据所述目标参数进行所述等效卷积运算时,具体用于:
利用所述寄存器中的配置信息进行等效卷积运算,得到所述等效卷积运算的结果。
[0026]根据本公开实施例的第二方面,提供一种芯片,包括第一方面所述的运算装置。
[0027]根据本公开实施例的第三方面,提供一种反卷积运算方法,包括:
[0028]获取反卷积运算的原始参数;
[0029]根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数;
[0030]利用卷积运算单元,根据所述目标参数进行所述等效卷积运算,并将所述等效卷积运算的结果确定为所述反卷积运算的结果。
[0031]在一个实施例中,所述原始参数包括下述至少一项:
[0032]原始输入数据、原始卷积核、原始卷积步长、原始填充值以及输出数据的尺寸。
[0033]在一个实施例中,所述根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数,包括:
[0034]根据所述原始输入数据的尺寸、所述原始卷积核的尺寸、所本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种运算装置,其特征在于,包括:控制单元,用于获取反卷积运算的原始参数,根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数,并将所述目标参数发送至卷积运算单元,以及将所述等效卷积运算的结果确定为所述反卷积运算的结果;卷积运算单元,用于根据所述目标参数进行所述等效卷积运算,并将所述等效卷积运算的结果发送至所述控制单元。2.根据权利要求1所述的运算装置,其特征在于,所述原始参数包括下述至少一项:原始输入数据、原始卷积核、原始卷积步长、原始填充值以及输出数据的尺寸。3.根据权利要求2所述的运算装置,其特征在于,所述目标参数包括目标输入数据;所述控制单元用于根据所述反卷积运算的原始参数,确定所述反卷积运算对应的等效卷积运算的目标参数时,具体用于:根据所述原始输入数据的尺寸、所述原始卷积核的尺寸、所述原始卷积步长、所述原始填充值以及所述输出数据的尺寸,确定所述等效卷积运算的目标输入数据的尺寸,以及所述原始输入数据中的每个原始元素在所述目标输入数据中的位置;并根据所述原始输入数据中每个原始元素在所述目标输入数据中的位置,将所述原始输入数据中的每个原始元素添加到所述目标输入数据的对应位置,并在所述目标输入数据中的其他位置添加预设值,得到所述目标输入数据。4.根据权利要求3所述的运算装置,其特征在于,所述芯片还包括矢量运算单元和存储单元;所述控制单元用于根据所述原始输入数据中每个原始元素在所述目标输入数据中的位置,将所述原始输入数据中的每个原始元素添加到所述目标输入数据的对应位置,并在所述目标输入数据中的其他位置添加预设值,得到所述目标输入数据时,具体用于:根据所述目标输入数据的尺寸,在存储单元上获取对应尺寸的目标数据空间,并将所述目标数据空间中的每个位置上的目标元素初始化为所述预设值;并根据所述原始输入数据中的每个原始元素在所述目标输入数据中的位置,确定每个原始元素在所述目标数据空间中的位置;以及将所述原始输入数据的地址和每个原始元素在所述目标数据空间中的位置发送至所述矢量运算单元;所述存储单元用于:存储所述目标输入数据;所述矢量运算单元用于:根据所述原始输入数据的地址获取所述原始输入数据中的每个原始元素,并使用每个原始元素对所述目标数据空间中的对应位置上的目标元素进行更新。5.根据权利要求4所述的运算装置,其特征在于,所述控制单元用于将所述原始输入数据的地址和每个原始元素在所述目标数据空间中的位置发送至所述矢量运算单元时,具体用于:根据所述原始输入数据的地址、所述目标数据空间的地址、以及所述每个原始元素在所述目标数据空间中的位置,确定所述矢量运算单元的配置信息,并将所述配置信息发送至所述矢量运算单元,其中,所述矢量运算单元的配置信息包括下述至少一项:输入数据的首地...

【专利技术属性】
技术研发人员:勾志宏胡英俊徐宁仪田志仲
申请(专利权)人:上海阵量智能科技有限公司
类型:发明
国别省市:

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

1