Fix typing for `setup_context` in `autograd` (#101464)
The original only matches a tuple of length 1, but it's intended to match any length.
Also, it now aligns with the docstring at L320
https://github.com/pytorch/pytorch/blob/d5cba0618aa0441af17487dd24317cd62afd0528/torch/autograd/function.py#L320
Pull Request resolved: https://github.com/pytorch/pytorch/pull/101464
Approved by: https://github.com/soulitzer, https://github.com/kit1980