pytorch

73f4c1a4 - [reland2] Update custom Function preserve torch function when inputs … (#110895)

[reland2] Update custom Function preserve torch function when inputs … (#110895) …returned as-is Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/110895 Approved by: https://github.com/albanD