一种池化算子并行计算方法、装置、电子设备、存储介质及产品

本发明公开了一种池化算子并行计算方法、装置、电子设备、存储介质及产品。该方法包括：响应于池化算子并行计算事件被触发，获取机器学习模型中的池化算子的各个通道的输入特征图和池化参数；针对每个通道的输入特征图，确定所述输入特征图的第一尺寸信息，并根据所述第一尺寸信息及所述池化参数确定在将所述输入特征图输入至所述池化算子后对应的输出特征图的第二尺寸信息；根据所述第二尺寸信息确定所述池化算子对所述输入特征图的并行计算策略；控制所述池化算子基于所述并行计算策略对所述输入特征图进行池化操作。本方案实现了池化算子的高效并行计算，不仅可有效避免负载不均衡的问题，而且还使得机器学习模型的整体计算性能有较大提升。