使用之前使用的 temp
temp.max(axis=0)
array([1.1, 1.1, 1.1, 1.1])
也可以这样
np.max(temp, axis=-1)
还可以这样
np.argmax(temp, axis=-1)
array([0, 2, 1, 3])
Last updated 2 years ago