tensorflow中取值

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

转载请标明出处:http://blog.csdn.net/wuzqchom/article/details/75281544


最近在写用tensorflow的程序时,中途遇到想取出tensorflow中的返回值是什么,可是其返回值也是一个tensor。用了两种方法,试图将tensor直接转为变量类型:

tf.cast(value,dtype=int)
tf.to_int32(value)

可是,着两条语句返回的都是一个tensor的信息,最后找到了这篇博客,两种方法都可以得到同样的结果:


import tensorflow as tf
a = tf.constant(1.)
b = tf.constant(6.)
c = a*b

with tf.Session() as sess:
    value_float = sess.run(c)
    print(type(c.eval()),c.eval())
    print(type(value_float),value_float)

输出:

<class 'numpy.float32'> 6.0
<class 'numpy.float32'> 6.0

之后我们就可以愉快的玩耍了。
tf.Variable也可以用同样的方式获得,输出结果与前一致:

import tensorflow as tf
a = tf.Variable(1.)
b = tf.Variable(6.)
c = tf.multiply(a,b)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    value_float = sess.run(c)
    print(type(c.eval()),c.eval())
    print(type(value_float),value_float)

猜你喜欢

转载自blog.csdn.net/wuzqChom/article/details/75281544