tensorflow-argmax

Posted 2021-01-21

import numpy as np
x=[1,3,4]
x1=np.equal(x,4)
x2=np.equal(x,3)
x4=np.equal(x,1)
z=np.stack(np.int32([x1,x2,x4]))
print z
print np.argmax(z)

[[0 0 1]
 [0 1 0]
 [1 0 0]]
2

import tensorflow as tf
x=[1,3,4]
x1=tf.equal(x,4)
x2=tf.equal(x,3)
x4=tf.equal(x,1)
z=tf.stack(tf.to_int32([x1,x2,x4]))
with tf.Session() as sess:
    print sess.run(z)
    print sess.run(tf.argmax(z))

[[0 0 1]
 [0 1 0]
 [1 0 0]]
[2 1 0]