Writing Custom Python Layer With Learnable Paramet

问题:

I know that this example is supposed to illustrate how to add trainable parameters in a Python layer using the add_blob() method.

However, I am still unable to understand how this can be used to set the dimensions of the blob based on user defined parameters.

There is a better example here on how to write a Python layer here. But here, the layer does not contain any trainable parameters.

Please explain how to write a custom Python layer with trainable parameters.

回答1:

When you add a parameters blob using add_blob(), you can reshape the added blob, either in setup() method (right when you add it), or in the layer's reshape() method.