如何在张量流中打印张量形状?
问题描述:
如何给出批量输入打印张量的形状?下面的代码不起作用如何在张量流中打印张量形状?
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,:,:,:]})
答
首先,你没有定义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))
答
我做了以下内容:
x_in = tf.identity(x_)
with tf.Session() as sess:
print sess.run(tf.shape(x_in))
如果你是不是在找这一点,那么请给我们一些背景。