net=nn.Sequential(conv_1, conv_2, conv_3, conv_4, conv_5, nn.AdaptiveAvgPool2d((1,1)), nn.Flatten(), nn.Linear(512,10)) # 觀察模型各層的輸出尺寸 X=torch.rand(size=(1,1,224,224)) forlayerinnet: X=layer(X) print(layer.__class__.__name__,'output shape:\t', X.shape...
www.dbjr.com.cn/article/2470...htm 2025-6-7