TensorFlow에서 특정 축에서 텐서를 재 형성하는 방법은 무엇입니까?

Question

지금 난 모양 텐서가 (3 * 2, 2)

처럼 그리고 난 (3 2 * 2) 모양으로 모양을 변경하려면 다음과 같은 특정 축 :

어떻게해야합니까? 기본 tf.reshape()은

솔루션로 바꿀 것이다 : 나는 tensorflow과 tf.concat (에서 그 사용 slice 발견)을 하위 텐서 슬라이스 및 해결을 CONCAT 수 problem.You를 해결할 수 있습니다 내 문제가 정확히

Answer 1

나는 다음 코드를 시도하고 필요한 결과를 얻었다. 하지만 단계 수를 줄일 수 있는지 여부는 확실하지 않습니다.

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) 

희망이 있습니다.