包含参数和梯度的数值,可以分别通过data函数和grad函数来访问
collect_params
获取net变量所有嵌套(例如通过add函数嵌套)的层所包含的所有参数。它返回的同样是一个由参数名称到参数实例的字典.
这个函数可以通过正则表达式来匹配参数名,从而筛选需要的参数
包含参数和梯度的数值,可以分别通过data函数和grad函数来访问
collect_params
获取net变量所有嵌套(例如通过add函数嵌套)的层所包含的所有参数。它返回的同样是一个由参数名称到参数实例的字典.
这个函数可以通过正则表达式来匹配参数名,从而筛选需要的参数
本文标题:Parameter类
本文链接:https://www.haomeiwen.com/subject/pddtyhtx.html
网友评论