[发明专利]卷积计算方法、装置、计算机设备及存储介质

权利要求书

1.一种卷积计算方法，其特征在于，包括：

根据输入数据和输入通道数量，生成对应的输入数据存储规则，并基于所述输入数据存储规则，存储输入数据；

根据所述输入数据存储规则，按照所述输入通道和输出通道的对应关系，存储权重数据；

按照行数据的顺序，基于所述输入数据和所述权重数据进行卷积计算，得到每个输出通道的行卷积结果；

对每个所述输出通道的行卷积结果进行累加计算，得到卷积计算结果。

2.根据权利要求1所述的卷积计算方法，其特征在于，所述根据输入数据和输入通道数量，生成对应的输入数据存储规则，并基于所述输入数据存储规则，存储输入数据包括：

获取所述输入数据的行数据信息和输入通道信息，将行数据的顺序作为第一优选级，将输入通道的顺序作为第二优选级，并将所述第一优先级和所述第二优选级作为输入数据存储规则；

基于所述输入数据存储规则，按照行数据的顺序存储每一行的输入数据，其中，针对同一行数据，按照输入通道的顺序存储每个输入通道的行输入数据。

3.根据权利要求1所述的卷积计算方法，其特征在于，所述根据所述输入数据存储规则，按照所述输入通道和输出通道的对应关系，存储权重数据包括：

基于所述输入数据存储规则，获取输入数据存储方式；

基于所述输入数据存储方式，按照输入通道的顺序，生成权重数据矩阵，并基于所述权重数据矩阵进行所述权重数据的存储。

4.根据权利要求1所述的卷积计算方法，其特征在于，所述按照行数据的顺序，基于所述输入数据和所述权重数据进行卷积计算，得到每个输出通道的行卷积结果包括：

为每一行的行输入数据分配对应的卷积计算周期，在每个所述卷积计算周期中，按照输出通道的顺序，计算得到每个输出通道的行卷积结果。

5.根据权利要求4所述的卷积计算方法，其特征在于，所述卷积计算周期包括多个计算子周期，所述为每一行输入数据分配对应的卷积计算周期，在每个所述卷积计算周期中，按照输出通道的顺序，计算得到每个输出通道的行卷积结果包括：

在每个计算子周期中，按照输入通道的顺序，计算得到每个输出通道的行卷积子结果；

针对同一输出通道，对所有的所述行卷积子结果进行累加计算，得到每个输出通道的行卷积结果，并将所述行卷积结果进行缓存。

6.根据权利要求1所述的卷积计算方法，其特征在于，所述对每个所述输出通道的行卷积结果进行累加计算，得到卷积计算结果包括：

针对同一个输出通道，获取每个行数据的行卷积结果，并对所有的所述行卷积结果进行累加计算，得到输出通道卷积结果；

获取每个输出通道的所述输出通道卷积结果，并对所有输出通道卷积结果进行累加计算，得到所述卷积计算结果。

7.一种卷积计算装置，其特征在于，包括：

输入数据存储模块，用于根据输入数据和输入通道数量，生成对应的输入数据存储规则，并基于所述输入数据存储规则，存储输入数据；

权重数据存储模块，用于根据所述输入数据存储规则，按照所述输入通道和输出通道的对应关系，存储权重数据；

行卷积结果计算模块，用于按照行数据的顺序，基于所述输入数据和所述权重数据进行卷积计算，得到每个输出通道的行卷积结果；

卷积结果生成模块，用于对每个所述输出通道的行卷积结果进行累加计算，得到卷积计算结果。

8.根据权利要求7所述的卷积计算装置，其特征在于，所述输入数据存储模块包括：

数据存储规则生成单元，用于获取所述输入数据的行数据信息和输入通道信息，将行数据的顺序作为第一优选级，将输入通道的顺序作为第二优选级，并将所述第一优先级和所述第二优选级作为输入数据存储规则；

数据存储单元，用于基于所述输入数据存储规则，按照行数据的顺序存储每一行的输入数据，其中，针对同一行数据，按照输入通道的顺序存储每个输入通道的行输入数据。