Parametrizations depending on several inputs (#60530)
Summary:
Resubmit of https://github.com/pytorch/pytorch/pull/58488
There was a line that had been changed in `test_nn.py` as caught in https://github.com/pytorch/pytorch/pull/58488#discussion_r651267668
I reverted that line, which should never have been changed. I reckon that should solve the issue.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60530
Reviewed By: ngimel
Differential Revision: D29329865
Pulled By: albanD
fbshipit-source-id: 8dfd0cd968fe26a3924dae7ca366af2c8a8639b3
