pytorch

3036a030 - [skip ci]Revert "Add comments for adding shape function and linting"

[skip ci]Revert "Add comments for adding shape function and linting" This is a technical revert of 6d36bbde7eb2eb0aed448f694338cb49c2ae47f3 to reconcile it with e50478c02592597f12b8490ec5496f76c7d8b8cc (which is the same + lint changes applied) Should be skipped during import