代码中的错误：模块“torch.nn”没有属性“ConvTranspose2D”

Question

def make_gen_block(self,input_channels,output_channels,kernel_size=3,stride=2,final_layer = False): 10 如果不是 Final_layer ： ---> 11 返回 nn.Sequential(nn.ConvTranspose2D(input_layer,output_layer,kernel_size,stride), 12 nn.BatchNorm2d（输出通道）， 13 nn.ReLU(inplace = True),)

AttributeError：模块“torch.nn”没有属性“ConvTranspose2D” 解决办法是什么？

在 pytorch 中转置

Answer 1

应该是

nn.ConvTranspose2d

我也有同样的问题。更多关于 PyTorch 文档