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

Question

我有一个torch.Size([1, 64, 8, 32, 32])，我希望在转置3d卷积后成为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))