pytorch

85839c4a - refactor according ed's feedback