Tensorflow中显示shape的真值

本文主要是介绍Tensorflow中显示shape的真值，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Tensorflow中显示shape的真值

a = tf.constant([[3, 4]], dtype=tf.float32)
可不定义,没有小数点int32,有小数点float32
print(a)
print(tf.shape(a))
这个就是对shape[1,2]的print [1,2]的操作名称是Shape:0,shape是(2),数据类型是Int32

with tf.Session() as sess:
print(sess.run(tf.shape(a)))
# 这样就可以输出[1 2]

这篇关于Tensorflow中显示shape的真值的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---