liumlau的博客

原创 关于numpy中的广播与argmax问题

问题背景 最近正在看faster-rcnn的anchor-target-layer.py源码，其中关于numpy的argmax（）问题搞得很混乱。此问题理清楚还需要高清楚numpy的索引的运作原理。 a = np.array([[1,3,5],[2,2,9]]) >>> a.argmax(axis=0) array([1, 0, 1]) 使用argmax（axis