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的主要内容,如果未能解决你的问题,请参考以下文章

VSCode自定义代码片段——CSS选择器

谷歌浏览器调试jsp 引入代码片段,如何调试代码片段中的js

片段和活动之间的核心区别是啥?哪些代码可以写成片段?

VSCode自定义代码片段——.vue文件的模板

VSCode自定义代码片段6——CSS选择器

VSCode自定义代码片段——声明函数