[发明专利]一种优化填充参数的方法、装置、计算机可读存储介质

权利要求书

1.一种优化填充参数的方法，包括：

根据输入数据计算输出数据的输出尺寸；

根据所述输出尺寸和所述输入数据计算所述输入数据的有效的输入尺寸；

根据所述有效的输入尺寸与所述输入数据中相应参数之间的比较计算优化后的填充参数。

2.根据权利要求1所述的优化填充参数的方法，所述输入数据中相应参数包括：

所述输入数据的输入尺寸、填充参数、卷积核、卷积核扩张参数和步长。

3.根据权利要求2所述的优化填充参数的方法，还包括：

根据所述优化后的填充参数和所述输入数据得到具有优化后填充参数的输入数据，并且输出所述具有优化后填充参数的输入数据用于卷积运算。

4.根据权利要求3所述的优化填充参数的方法，其中所述根据输出尺寸和所述输入数据计算所述输入数据的有效的输入尺寸包括：将所述输出尺寸减一得到的数值与所述步长乘积，再加上所述卷积核扩张后的尺寸，从而得到所述输入数据的有效的输入尺寸。

5.根据权利要求4所述的优化填充参数的方法，其中所述根据有效的输入尺寸与所述输入数据中相应参数之间的比较计算优化后的填充参数包括：

判断所述输入尺寸和所述填充参数之和与所述有效的输入尺寸之间的大小关系，并且根据所述大小关系计算优化后的填充参数。

6.根据权利要求5所述的优化填充参数的方法，其中所述判断输入尺寸和所述填充参数之和与所述有效的输入尺寸之间的大小关系，并且根据所述大小关系计算优化后的填充参数包括：

在所述输入尺寸和填充参数之和小于所述有效的输入尺寸的情况下，对应维度卷积滑动起始端补零量与所述填充参数相同，对应维度卷积滑动结束端补零量等于所述有效的输入尺寸减去所述填充参数与输入数据的输入尺寸的和。

7.根据权利要求5所述的优化填充参数的方法，其中所述判断输入尺寸和所述填充参数之和与所述有效的输入尺寸之间的大小关系，并且根据所述大小关系计算优化后的填充参数包括：

在所述输入尺寸和填充参数之和大于等于所述有效的输入尺寸的情况下，对应维度卷积滑动起始端补零量与所述填充参数相同，对应维度卷积滑动结束端补零量等于零。

8.根据权利要求6或7所述的优化填充参数的方法，其中根据所述优化后的填充参数和所述输入数据得到具有优化后填充参数的输入数据包括：

使用所述优化后的填充参数对所述输入数据进行填充。

9.根据权利要求8所述的优化填充参数的方法，其中所述优化填充参数的方法是用于Pytorch框架的。

10.一种用于优化填充参数的装置，包括：

处理器，其配置用于执行程序指令；以及

存储器，其配置用于存储所述程序指令，当所述程序指令由所述处理器加载并执行时，使得所述装置执行根据权利要求1-9的任意一项所述的方法。

11.一种计算机可读存储介质，其中存储有程序指令，所述程序指令适于由处理器加载并执行根据权利要求1-9的任意一项所述的方法。