pytorch

a3478491 - changed another assert to check