Is it possible to use this library for CNN networks implemented with pytorch? Can you show me an example?