Solved by albanD in post #2 
              
              
                From the beginning of the stack trace, it seems like you’re trying to make a Tensor which is too large (600MB). Is that expected? How much memory does it use if you run it on your machine?