Tensorflow tf.contrib.layers.flatten(x)与tf.reformate(x,[n,1])相同吗?

Tensorflow tf.contrib.layers.flatten(x)与tf.reformate(x,[n,1])相同吗?,tensorflow,Tensorflow,我只是想知道tf.contrib.layers.flatte(x)是否与tf.reformate(x[n,1])相同?有人对此有想法吗?答案就在这里。它相当于以下内容: flattened = tf.reshape(x, [tf.shape(x)[0], -1]) 展平折叠除批次大小之外的所有维度[bs,n,m,k]->[bs,n*m*k]

我只是想知道
tf.contrib.layers.flatte(x)
是否与
tf.reformate(x[n,1])
相同?有人对此有想法吗?

答案就在这里。它相当于以下内容:

flattened = tf.reshape(x, [tf.shape(x)[0], -1])

展平折叠除批次大小之外的所有维度<代码>[bs,n,m,k]->[bs,n*m*k]