在Caffe中编写具有可学习参数的自定义Python层

Question

我知道this example应该说明如何使用add_blob()方法在Python layer中添加可训练参数。

但是，我仍然无法理解如何根据用户定义的参数设置blob的尺寸。

这里有一个关于如何编写Python层here的更好的例子。 但在这里，该图层不包含任何可训练的参数。

请解释如何使用可训练参数编写自定义Python图层。

Answer 1

使用add_blob()添加参数blob时，您可以reshape方法添加blob，setup()方法（添加时），或者在图层＆＃39; s reshape()方法。