欢迎光临散文网 会员登陆 & 注册

第十九篇—自定义层

2023-02-16 19:53 作者:YouOnly_LiveOnce  | 我要投稿


部分一:不含模型参数的自定义层

设计一个网络层,实现将输入减掉均值后再输出,代码如下:

可以将自定义层加入Sequential,构造更复杂的网络:



部分二:含模型参数的自定义层

通过ParameterList()实现:

结果:

可以看到,ParameterList接收一个Parameter实例的列表作为输入,再得到一个参数列表;

通过ParameterDict()实现:

结果:

可以看到,ParameterDict()接收一个Parameter实例的字典作为输入,得到一个参数字典;

关于字典常用的几个函数:

keys():返回键;

values():返回值;

items():返回键和值;

update():向字典中新增元素;

第十九篇—自定义层的评论 (共 条)

分享到微博请遵守国家法律