pytorch

5a48d385 - minor clarifications in comment