Как преобразовать тензор из [16,96,96,16] в [16,9216,16] в TensorFlow? - PullRequest
0 голосов
/ 18 апреля 2020

Я просто использую TensorFlow 1.8.0 для проектирования сети. Существует тензор с формой A = [16,96,96,16] (первые 16 - размер пакета, 96 - высота / ширина блока изображения, последние 16 - число карт характеристик)

Я бы хотел преобразовать форму тензора A в [16,9216,16] (где 9216 = 96x96), чтобы каждая карта объектов была преобразована в столбец из матрицы.

Могу ли я узнать, как реализовать это с помощью функции `tf.reshape 'в TensorFlow?

Заранее спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...