tensorflow的argmax和numpy的argmax一样:
import tensorflow as tf
import numpy as np
a = np.array([[1,2,3],[1,2,3],[9,8,7]])
mm = tf.argmax(a,1)
sess = tf.Session()
print sess.run(mm)
bb = np.argmax(a,1)
print bb
输出:
[2 2 0]
[2 2 0]
tensorflow的argmax和numpy的argmax一样:
import tensorflow as tf
import numpy as np
a = np.array([[1,2,3],[1,2,3],[9,8,7]])
mm = tf.argmax(a,1)
sess = tf.Session()
print sess.run(mm)
bb = np.argmax(a,1)
print bb
[2 2 0]
[2 2 0]