包含參數(shù)和梯度的數(shù)值沥阱,可以分別通過data函數(shù)和grad函數(shù)來訪問
collect_params
獲取net變量所有嵌套(例如通過add函數(shù)嵌套)的層所包含的所有參數(shù)。它返回的同樣是一個(gè)由參數(shù)名稱到參數(shù)實(shí)例的字典.
這個(gè)函數(shù)可以通過正則表達(dá)式來匹配參數(shù)名颗圣,從而篩選需要的參數(shù)
包含參數(shù)和梯度的數(shù)值沥阱,可以分別通過data函數(shù)和grad函數(shù)來訪問
collect_params
獲取net變量所有嵌套(例如通過add函數(shù)嵌套)的層所包含的所有參數(shù)。它返回的同樣是一個(gè)由參數(shù)名稱到參數(shù)實(shí)例的字典.
這個(gè)函數(shù)可以通過正則表達(dá)式來匹配參數(shù)名颗圣,從而篩選需要的參數(shù)