在Caffe中編寫具有可學習參數的自定義Python層

Question

我知道這個例子應該說明如何使用add_blob()方法在Python層中添加可訓練參數。

但是，我仍然無法理解如何根據用戶定義的參數設置blob的尺寸。

有關於如何編寫一個Python層在這里一個更好的例子在這里 。 但是在這里，該層不包含任何可訓練的參數。

請解釋如何編寫具有可訓練參數的自定義Python層。

Answer 1

使用add_blob()添加參數blob時，可以在setup()方法（添加時）或圖層的reshape()方法中reshape添加添加的blob。