如何知道输入X的形状,它是TF中的占位符

时间:2017-08-20 07:50:00

标签: tensorflow

我有代码:

X = tf.placeholder(dtype=tf.float32, shape=[None, 28, 28, 1])
W = tf.variable...
....
Z ...

Z是张量并从X计算,这里我想知道Z的批量大小,即Z的第一个元素值,我试过了     Z.get_shape()[0] .value和tf.shape(Z)[0]。两个结果都是无,如何解决?

提前致谢。

1 个答案:

答案 0 :(得分:0)

这是一个如何获得它的例子:

z = tf.Variable(tf.zeros(shape=(2, 3, 4)))
with tf.Session() as sess:
    print sess.run(tf.shape(z))

因为TF中的所有内容都在会话中,所以您需要运行会话来执行您想要运行的功能!