모양 (n, h, w, c)에서 (n, h, w, c)까지 모양 변경

Question

Best way to flatten a 2D tensor containing a vector in TensorFlow? 나는 TensorFlow에서 2D 텐서를 1D 텐서로 평탄화하는 방법을 보았지만 4D TensorFlow 기능을 갖춘 3D 텐서에 텐서? tf.reshape() 기능은 당신을 위해 일한다

import tensorflow as tf 
X = tf.random_normal([n, h, w, c], mean=1, stddev=4) 
Y = ??? 

Answer 1

:

Y = tf.reshape(X, (n, h*w, c)) 

예를 들어