2015-12-10 2 views
21

Каков наиболее эффективный способ сглаживания 2D-тензора, который на самом деле является горизонтальным или вертикальным вектором в 1D-тензор?Лучший способ сгладить 2D-тензор, содержащий вектор в TensorFlow?

Есть ли разница в плане производительности между:

tf.reshape(w, [-1]) 

и

tf.squeeze(w) 

?

ответ

31

Оба «tf.reshape(w, [-1])» и «tf.squeeze(w)» являются «дешевыми», поскольку они действуют только на метаданные (т. Е. Форму) данного тензора и не изменяют сами данные. Из двух tf.reshape() имеет немного simpler logic internally, но исполнение двух должно быть неразличимым.

+0

См. [Это сообщение] (https://stackoverflow.com/questions/36668542/flatten-batch-in-tensorflow) для выравнивания при сохранении размера партии. – Alex