2017-02-24 71 views

回答

1

濾波器權重位於卷積的weight參數中。假設你已經使用了標準佈局,就像你的例子中那樣,權重將具有形狀(num_filter,channels,kernel [0],kernel [1])。

例如

conv1.weight.data()[0] 

訪問重量張量從當前上下文中的第一過濾器的。

相關問題