2017-10-23 10 views
0

は今、私は次のような特定の軸で(2 * 2、3)TensorFlowでは、特定の軸のテンソルをどのように変形するのですか?

enter image description here

のように見えると私は形でそれを再構築したい(2、3 * 2)形状を持つテンソルを持っています。

enter image description here

私は何をすべき?デフォルトtf.reshape()

enter image description here

SOLUTIONにそれを再構築します:私はtensorflowとtf.concat(でその使用sliceを発見した)を、サブテンソルをスライスし、解決するためにそれらをCONCATできproblem.Youを解決することができ、私の問題が正確に

答えて

0

私は次のコードを試して、必要な結果を得ました。しかし、ステップ数を減らすことができるかどうかは分かりません。

import tensorflow as tf 

x = [[1, 2], 
    [3,4], 
    [5,6], 
    [7,8], 
    [9,10], 
    [11,12]] 

a = tf.reshape(x,[-1,6]) 
b = tf.split(a,3, 1) 
c = tf.reshape(b,[-1,4]) 

X=tf.placeholder(tf.float32, shape=[6, 2], name='input') 

with tf.Session() as sess: 
     c = sess.run(c, feed_dict={X: x}) 
     print(c) 

これが役に立ちます。

+0

はい、うまくいきますが、tensorflowとtf.concatの 'slice'が私の問題を解決することができます。とにかく助けてくれてありがとうございます:) – FesianXu

関連する問題