eppoj

@karanjakhar can you give me a heads up when you got this working?

Thank you that was the problem (learned something). However now I'm running into memory issues (OOM when allocating tensor with shape). I'm trying to run the software on an nvidia...

@karanjakhar where should I execute this code ?