xh929175050的博客

原创 np.argmax输出的是什么？

重点说一下 argmax输出的是什么东西test = np.array([[1, 2, 3], [2, 3, 4], [5, 4, 3], [8, 7, 2]]) # 构建一个4X3的矩阵out = np.argmax(test, axis=1) # axis=1：按行查找最大元素 axis...