Блог

Главная - Вопросы по программированию - каков простой способ изменить форму тензора с [batch_size, 64, 3, 17] на [batch_size, 1024, 7, 7]

каков простой способ изменить форму тензора с [batch_size, 64, 3, 17] на [batch_size, 1024, 7, 7]

#deep-learning #pytorch #conv-neural-network

Вопрос:

Я хочу объединить две модели. первая форма вывода [batch_size, 64, 3, 17], а вторая форма ввода модели [batch_size, 1024, 7, 7]. может быть, фильтр через другой слой CNN может помочь мне получить эту форму?

Метки: , , , , ,