как получить torch.Size ([1, 3, 16, 112, 112]) вместо torch.Size ([1, 3, 16, 64, 64]) после ConvTranspose3d

Question

У меня есть torch.Size([1, 64, 8, 32, 32]), который я хочу, чтобы после моей транспонированной трехмерной свертки он стал torch.Size([1, 3, 16, 112, 112]).

Используя это: nn.ConvTranspose3d(64, 3, kernel_size=4, stride=2, bias=False, padding=(1, 1, 1)) Я получаю правильные выходные каналы и количество кадров, но не размеры кадров: torch.Size([1, 3, 16, 64, 64])

Что я должен изменить, чтобы выбрать правильный факел. Размер?

Answer 1

Вы должны использовать разные шаги и отступы для разных размеров.

ConvTranspose3d(64, 3, kernel_size=4, stride=(2, 4, 4), bias=False, padding=(1, 8, 8))