2017-10-10 13 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)) 

あなたはこのために見ていない場合は、私たちにいくつかのコンテキストを与えてください。

関連する問題