diffpool程序理解

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

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

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

3.torch.cat(A,B,dim=0)将A、B张量拼接在一起，0按行拼接（列对齐）；1按列拼接（行对齐）

4.nn.BatchNorm1d():

BatchNorm对训练过程有着更根本的影响：它能使优化问题的解空间更加平滑，而这种平滑性确保了梯度更具预测性和稳定性，因此可以使用更大范围的学习速率并获得更快的网络收敛。

5.torch.matmul(A,B)张量相乘