我找到了以下关于使用 skimage 的“numpy”和“block_reduce”实现最大池的答案。但它没有显示如何获取最大值的索引(最好是相对于实际矩阵,而不是池化矩阵)。什么是计算速度最快的解决方案?
import numpy as np
import skimage.measure
a = np.array([
[ 20, 200, -5, 23],
[ -13, 134, 119, 100],
[ 120, 32, 49, 25],
[-120, 12, 9, 23]
])
skimage.measure.block_reduce(a, (2,2), np.max)