如何在tensorflow中打印张量形状？

Question

如何在批量输入的情况下打印张量的形状？以下代码不起作用

x_in = tf.identity(x_)
print_x_in = tf.Print(x_in, x_in.get_shape())

init = tf.global_variables_initializer()

# Start a new TF session
sess = tf.Session()

# Run the initializer
sess.run(init)

# feed in batch
sess.run(x_in, feed_dict={x_: x[1:10,:,:,:]})

Answer 1

首先，您没有定义x_。您需要一个占位符，与

一致

x_ = tf.placeholder(shape=[None, shape[0],shape[1],shape[2],dtype=tf.float32)

然后你可以输入x的值为x _。

进入会话后，您将评估张量

x_out = sess.run(x_in, feed_dict={x_: x[1:10,:,:,:]})

然后你可以打印。

print(np.shape(x_out))

Answer 2

我执行以下操作：

x_in = tf.identity(x_)
with tf.Session() as sess:
    print sess.run(tf.shape(x_in))

如果你不是在寻找这个，那么请给我们一些背景信息。