pytorch

66519916 - oops, change test to actually stress the right thing