PyTorch implementation

[Sparkier]

Describe the request
It would be nice to also make Net2Vis available for pytorch users.

How can it be tackled?
We can likely implement something similar to the keras translate with model.state_dict().

[Sparkier]

I tried implementing something like this with this feature based on onnx conversion: https://github.com/viscom-ulm/Net2Vis/tree/feature/torch
However, there are multiple problems with respect to performance and onnx conversion. I currently do not have the time to debug this.