标签: tensorflow
我只是想知道tf.contrib.layers.flatten(x)是否与tf.reshape(x, [n, 1])相同?有人有这方面的想法吗?
tf.contrib.layers.flatten(x)
tf.reshape(x, [n, 1])
答案 0 :(得分:2)
implementation的tf.layers.flatten()揭示了答案。它等同于以下内容:
tf.layers.flatten()
flattened = tf.reshape(x, [tf.shape(x)[0], -1])