Hello,
I am trying to create some layers. Let we said convolutional layer but will have some logic operation inside of it. Basically I can do that using tensor operation and create it as function inside the class. However I am so confuse how can I update the weight of my custom convolution filter?
As far as I know most of people implement the new layer and they will still using torch.nn.functional so how can I do this?
I have read some threads like here , here , also here 170 .
Or do we just use nn.Parameter is enough like in here .
I am still confuse where to start, suggestion to start is welcome.