我有代码:
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]。两个结果都是无,如何解决?
提前致谢。
答案 0 :(得分:0)
这是一个如何获得它的例子:
z = tf.Variable(tf.zeros(shape=(2, 3, 4)))
with tf.Session() as sess:
print sess.run(tf.shape(z))
因为TF中的所有内容都在会话中,所以您需要运行会话来执行您想要运行的功能!