machine-learning - 在 Caffe 中使用可学习参数编写自定义 Python 层

Question

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

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

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

请解释如何编写具有可训练参数的自定义 Python 层。

Answer 1

当您使用 add_blob() 添加参数 blob 时，您可以在 setup() 方法中 reshape 添加的 blob(右)当您添加它时)，或者在图层的 reshape() 方法中。