[发明专利]针对折叠特征数据执行卷积运算的方法和装置

权利要求书

1.一种针对折叠特征数据执行卷积运算的方法，包括：

对提供给所述卷积层的折叠特征数据和原始卷积核执行预处理；

根据所述折叠特征数据的折叠方式，在宽度和高度中的至少一个维度上折叠经预处理的原始卷积核，以生成与所述原始卷积核相对应的一个或多个折叠卷积核，其中在所述宽度和高度中的至少一个维度上的折叠受到执行卷积运算的硬件所支持的通道数量的限定；以及

使用所述一个或多个折叠卷积核对经预处理的折叠特征数据执行所述卷积运算。

2.根据权利要求1所述的方法，其中，所述折叠特征数据与在第一维度上未折叠的第一特征数据相对应，所述第一特征数据中的在所述第一维度上的第i_fx*N_x+j_fx个切片的所有C_x个通道的数据对应于所述折叠特征数据的在所述第一维度上的第i_fx个切片的从第j_fx*C_x个通道开始的连续的C_x个通道的数据，所述第一维度是宽度和高度中的一个，i_fx为大于或等于0的整数，N_x为大于1的整数，j_fx为大于或等于0且小于N_x的整数，C_x为大于0的整数。

3.根据权利要求2所述的方法，其中，所述预处理包括：

根据所述卷积层所指定的填充方式来确定第一填充量P₁，所述第一填充量P₁大于或等于0并且表示所述填充方式所要求的在第一特征数据的在所述第一维度上的起始边界处的填充量；

在所述折叠特征数据的在所述第一维度上的起始边界处填充个的零切片，表示向上取整运算；以及

在所述原始卷积核的在所述第一维度上的起始边界处填充个零切片。

4.根据权利要求2所述的方法，其中，执行所述卷积运算包括：

在所述原始卷积核的在所述第一维度上的第一步长不等于N_x的情况下，在使用与所述原始卷积核相对应的所有折叠卷积核对经预处理的折叠特征数据的同一部分进行卷积之后，在所述第一维度上按照所述第一步长移动与所述原始卷积核相对应的所有折叠卷积核。

5.根据权利要求2所述的方法，其中，执行所述卷积运算包括：

在所述原始卷积核的在所述第一维度上的第一步长不等于N_x并且分别使用与所述原始卷积核相对应的每个折叠卷积核对整个经预处理的折叠特征数据进行卷积的情况下，每个折叠卷积核的在所述第一维度上的步长等于所述第一步长。

6.根据权利要求2所述的方法，其中，执行所述卷积运算包括：

在所述原始卷积核的在所述第一维度上的第一步长等于N_x的情况下，每个折叠卷积核的在所述第一维度上的步长为1。

7.根据权利要求2至6中的任一项所述的方法，其中，在所述至少一个维度上折叠经预处理的原始卷积核包括：

在经预处理的原始卷积核的在所述第一维度上的起始边界处分别填充k_x*S_x个零切片，以生成E_x个第一变换卷积核，S_x为所述原始卷积核的在所述第一维度上的第一步长，E_x大于或等于1并且取决于N_x和S_x的值，k_x为大于等于0且小于E_x的整数；以及

以将所述第一维度上的每N_x个的连续的切片在深度上拼接到一起的方式，在所述第一维度上对每个第一变换卷积核进行的第一折叠，以针对每个第一变换卷积核生成相应的第一折叠卷积核。