Documentation

[levtelyatnikov]

I have found that all the wrappers in the doc have exactly the same row, see below

Please update it with the appropriate documentation

[levtelyatnikov]

1. I checked other modules, and most of the documentation is wrong. Please update the documentation. In most of the cases the issue is similar to the one I posted above. Use torch_geometric to get inspiration on making the documentation because we will use the documentation to generate the webpage later.

2. There is no consistency in what model gets as input during init, call, forward methods, and what is written in the documentation. Please update this as well.

3. Add repr method like in the image below to all classes. The brackets' parameters must represent init input.