我知道这个例子 https://github.com/BVLC/caffe/blob/master/python/caffe/test/test_python_layer.py#L31应该说明如何在Python层 https://stackoverflow.com/a/41481539/1714410使用add_blob() method.
但是，我仍然无法理解如何使用它来根据用户定义的参数设置 blob 的尺寸。
这里有一个关于如何编写 Python 层的更好示例here https://github.com/BVLC/caffe/blob/master/examples/pycaffe/layers/pyloss.py。
但在这里，该层不包含任何可训练的参数。
请解释如何编写具有可训练参数的自定义 Python 层。
当您使用添加参数 blob 时add_blob()， 你可以reshape添加的斑点，要么在setup()方法（当你添加它时），或者在图层的reshape() method.
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
