Python — (tf/np).argmax()

閱讀時間約 1 分鐘
tensorflow和numpy 中都有個argmax()
兩種用法都一樣(差別在tensorflow 要用sess.run()啟動)下面只展示numpy
用法如下
import numpy as np
arr = np.array([[1, 10, 3], [6, 5, 11], [7, 8, 9], [12, 2, 4]])
print('axis=0', np.argmax(arr, axis=0))
print('axis=1', np.argmax(arr, axis=1))# arr[0] = [1, 10, 3]
# arr[1] = [6, 5, 11]
# arr[2] = [7, 8, 9]
# arr[3] = [12, 2, 4]
結果
axis=0 [3 0 1]
axis=1 [1 2 2 0]
**當axis=0時,會比對在column中最大數字回傳它的索引值
**當axis=1時,會比對在row中最大數字回傳它的索引值
為什麼會看到廣告
John
John
留言0
查看全部
發表第一個留言支持創作者!