Make torch.library decorators return function
This allows for chaining.
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78996
Approved by: https://github.com/Lezcano, https://github.com/Chillee
