nn.Sequential的参数的注意事项

在Python中，*作用在形参上，代表这个位置接收任意多个非关键字参数，转化成元组方式；*作用在实参上，代表的是将输入迭代器拆成一个个元素。

从nn.Sequential的定义来看，输入要么是orderdict,要么是一系列的模型，遇到list，必须用*号进行转化，否则会报错 TypeError: list is not a Module subclass

# 下面的三种layer都是可以
layer = nn.Sequential(nn.Linear(10,20),nn.ReLU(),nn.Linear(20,5))
layer = (nn.Linear(10,20),nn.ReLU(),nn.Linear(20,5))
layer = [nn.Linear(10,20),nn.ReLU(),nn.Linear(20,5)]
net = nn.Sequential(*layer)
net.to(device=torch.device("cuda:0"))
summary(net,(1,20 ,10))