Parameter类

2020-03-17  本文已影响0人  elementmaster

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

collect_params

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

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

上一篇 下一篇

猜你喜欢

热点阅读