PyTorch的顺序容器——torch.nn.Sequential
发布时间
阅读量：
阅读量
1.torch.nn.Sequential概要
pytorch官方文档中对torch.nn.Sequential模块的功能进行了如下说明。
应用方法：
# 写法一
net = nn.Sequential(
nn.Linear(num_inputs, 1)
# 此处还可以传入其他层
)
# 写法二
net = nn.Sequential()
net.add_module('linear', nn.Linear(num_inputs, 1))
# net.add_module ......
# 写法三
from collections import OrderedDict
net = nn.Sequential(OrderedDict([
('linear', nn.Linear(num_inputs, 1))
# ......
]))
全部评论 (0)
还没有任何评论哟~