pytorch

f677bca7 - fix generation of pivot tensor and removal of nan