第十九篇—自定义层
2023-02-16 19:53 作者:YouOnly_LiveOnce | 我要投稿
部分一:不含模型参数的自定义层
设计一个网络层,实现将输入减掉均值后再输出,代码如下:
可以将自定义层加入Sequential,构造更复杂的网络:
部分二:含模型参数的自定义层
通过ParameterList()实现:
结果:

可以看到,ParameterList接收一个Parameter实例的列表作为输入,再得到一个参数列表;
通过ParameterDict()实现:
结果:

可以看到,ParameterDict()接收一个Parameter实例的字典作为输入,得到一个参数字典;
关于字典常用的几个函数:
keys():返回键;
values():返回值;
items():返回键和值;
update():向字典中新增元素;