Как объединить тензор вдоль неизвестного измерения (tf2) / полностью сверточной сети? - PullRequest
0 голосов
/ 25 апреля 2020

С учетом тензора X формы (batch_size, None, h, w, c) , где Нет указывает количество пятен изображения формы (ч , ш, c) . Количество пятен изображения варьируется (полностью сверточный параметр) и поэтому неизвестно.

Как выполнить горизонтальное выравнивание пятен изображения вдоль оси 3? Например, X формы (batch_size 5, h, w, c) - как связать тензор с (batch_size, h, w * 5, c).

Обратите внимание:

  • tf.reshape приводит к правильной выходной форме, однако изменяет порядок значений пикселей (я хочу выровнять патчи изображений рядом)
  • , если число патчей изображений было известно, это может быть достигнуто с помощью tf.split + tf.concat; или перебирая количество картинок патчей + tf.concat.

Спасибо.

...