python - Best way to flatten a 2D tensor containing a vector in TensorFlow?

Question

What is the most efficient way to flatten a 2D tensor which is actually a horizontal or vertical vector into a 1D tensor?

Is there a difference in terms of performance between:

tf.reshape(w, [-1])

and

tf.squeeze(w)

?

question from:https://stackoverflow.com/questions/34194151/best-way-to-flatten-a-2d-tensor-containing-a-vector-in-tensorflow

Answer 1

Both tf.reshape(w, [-1]) and tf.squeeze(w) are "cheap" in that they operate only on the metadata (i.e. the shape) of the given tensor, and don't modify the data itself. Of the two tf.reshape() has slightly simpler logic internally, but the performance of the two should be indistinguishable.