I’m not sure, what you mean by “giving parameters to there operations” exactly.
If you are not sure about whether to use an nn.Module or the functional API, have a look at this longer post, where I describe my personal point of view.
I’m not sure, what you mean by “giving parameters to there operations” exactly.
If you are not sure about whether to use an nn.Module or the functional API, have a look at this longer post, where I describe my personal point of view.