tensorflow随笔-argmax

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/u010255642/article/details/82945769

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]