tensorflow-argmax
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow-argmax相关的知识,希望对你有一定的参考价值。
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]
以上是关于tensorflow-argmax的主要内容,如果未能解决你的问题,请参考以下文章