Pytorch学习 -----nn.Sequential(*layers)


前言

学习Pytorch代码时,遇到了nn.Sequential(*layers),其中layers为list类型


一、*变量

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

二、nn.Sequential(*layers)

1.源码

nn.Sequential的定义来看,输入要么是orderdict,要么是一系列的模型

2.分析

nn.Sequential的定义来看,输入要么是orderdict,要么是一系列的模型
因为属于else类,由于*args转化为元组形式,list类型相当于1个元组元素,因此只能循环一次,加入的模型层却不止一个 ———-会报错

总结

遇到一个list传入,必须用*号进行转化

  • 3
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值