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
推荐阅读
- 危险也是机会
- 活着就是生命的全部意义
- 一个健康的APP和健全的人格大体类似
- NeuVector 会是下一个爆款云原生安全神器吗()
- 全过程工程咨询——时间管理(12)
- 别墅庭院设计,不同的别墅庭院设计也给人视觉上完全不一样的!
- 不让记忆、感觉、情绪成为孩子的负累|不让记忆、感觉、情绪成为孩子的负累|《全脑教养法》(四)
- (全员向连载)云间当铺(一)
- 「按键精灵安卓版」关于全分辨率脚本的一些理解(非游戏app)
- pytorch|使用pytorch从头实现多层LSTM