全连贯层
代码
import torchimport torch.nn as nnfc = nn.Linear(512, 10)input = torch.randn(1, 512, 1, 1)output = input.view(input.size(0), -1)output = fc(output)print(output.size())torch.Size([1, 10])援用
<1>
torch=1.7.1+cu101torchvision=0.8.2torchaudio=0.7.2