Numpy知识点：np.expand_dims()&np.argmax()