PyTorch_Linear(fc)

全连接层 【PyTorch_Linear(fc)】PyTorch_Linear(fc)
文章图片

PyTorch_Linear(fc)
文章图片

代码

import torch import 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+cu101 torchvision=0.8.2 torchaudio=0.7.2

    推荐阅读