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

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

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

tf.reshape(w, [-1])

и

tf.squeeze(w)

?

Ответы

Ответ 1

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