Понимание образца сетки Pytorch

Question

У меня есть входной тензор размером [1,32,296,400]

, и у меня есть набор пикселей [1, 56000, 400, 2]

После применения grid_sample с mode=‘bilinear’ У меня есть [1, 32, 56000, 400]

Могу ли я знать, что именно здесь произошло? Я знаю, что grid_sample предполагает эффективное преобразование пикселей в новое местоположение дифференцируемым образом, но эти измерения не проясняют, что происходит.

Answer 1

Пожалуйста, смотрите документацию grid_sample.

Ваш входной тензор имеет shape 1x32x296x400, то есть у вас есть один пример в пакете с 32 каналами и пространственные размеры 296x400 пикселей.
Кроме того, у вас есть «сетка» размером 1x56000x400x2, которую pytorch интерпретирует как новые местоположения для сетки пространственных измерений 56000x400, где каждое новое местоположение имеет координаты x, y, из которых можно выбрать новое значение сетки. Следовательно, информация «сетки» имеет форму 1x56000x400x2.
Выход, как и ожидалось, представляет собой 2D-тензор формы 1x32x56000x400: размеры пакета и канала не изменяются, но пространственные координаты соответствуют информации «сетки», предоставленной grid_sample.