pytorch

f46c73a4 - Test added for test_prod_gpu, under review, passes with changes successfully