Parameter类

包含参数和梯度的数值，可以分别通过data函数和grad函数来访问

collect_params

获取net变量所有嵌套（例如通过add函数嵌套）的层所包含的所有参数。它返回的同样是一个由参数名称到参数实例的字典.

这个函数可以通过正则表达式来匹配参数名，从而筛选需要的参数

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容