如何在ConvTranspose3d之后获得torch.Size（[1,3,16,112,112]）而不是torch.Size（[1,3,16,64,64]）

Question

在我转换3d卷积成为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

你应该使用不同的步幅和填充不同的dims。

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