diffpool程序理解

np.vstack()  将数组沿垂直方向堆叠起来

np.hstack()  将数组沿水平方向堆叠起来

numpy.argmax(a, axis=None, out=None)  返回沿轴axis最大值的索引号。axis=0按列输出，axis=1按行输出。