‘B’ is a NN with fully connected layers and convolutional layers, whose parameters are of type ‘Parameter’.
So, I could replace the usual layers by layers using torch.nn.functional functions… but I need to rewrite each layer I want to use, then. Is this the only solution?