tf.contrib.layers.flatten(x)是否与tf.reshape(x,[n,1])相同?

时间:2017-10-24 21:26:47

标签: tensorflow

我只是想知道tf.contrib.layers.flatten(x)是否与tf.reshape(x, [n, 1])相同?有人有这方面的想法吗?

1 个答案:

答案 0 :(得分:2)

implementationtf.layers.flatten()揭示了答案。它等同于以下内容:

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