如何在張量流中打印張量形狀？

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)) 

如果你是不是在找這一點，那麼請給我們一些背景。