La mejor manera de aplanar un tensor 2D que contiene un vector en TensorFlow?

Question

La mejor manera de aplanar un tensor 2D que contiene un vector en TensorFlow?

¿Cuál es la forma más eficiente de aplanar un tensor 2D que en realidad es un vector horizontal o vertical en un tensor 1D?

Hay una diferencia en términos de rendimiento entre:

tf.reshape(w, [-1])

Y

tf.squeeze(w)

?

26

python tensorflow

Author: Andrzej Pronobis, 2015-12-10

Source

1 answers

score 40 · Accepted Answer

Ambos tf.reshape(w, [-1]) y tf.squeeze(w) son "baratos" en el sentido de que operan solo en los metadatos (es decir, la forma) del tensor dado, y no modifican los datos en sí. De los dos tf.reshape()tiene un poco lógica más simple internamente, pero el rendimiento de los dos debe ser indistinguible.