2017-10-10 115 views
0

如何给出批量输入打印张量的形状?下面的代码不起作用如何在张量流中打印张量形状?

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,:,:,:]}) 

回答

2

首先,你没有定义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)) 
0

我做了以下内容:

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

如果你是不是在找这一点,那么请给我们一些背景。