第十九篇—自定義層
2023-02-16 19:53 作者:YouOnly_LiveOnce | 我要投稿
部分一:不含模型參數(shù)的自定義層
設計一個網(wǎng)絡層,實現(xiàn)將輸入減掉均值后再輸出,代碼如下:
可以將自定義層加入Sequential,構(gòu)造更復雜的網(wǎng)絡:
部分二:含模型參數(shù)的自定義層
通過ParameterList()實現(xiàn):
結(jié)果:

可以看到,ParameterList接收一個Parameter實例的列表作為輸入,再得到一個參數(shù)列表;
通過ParameterDict()實現(xiàn):
結(jié)果:

可以看到,ParameterDict()接收一個Parameter實例的字典作為輸入,得到一個參數(shù)字典;
關于字典常用的幾個函數(shù):
keys():返回鍵;
values():返回值;
items():返回鍵和值;
update():向字典中新增元素;