Add trace_backward, masked_select_backward, and take_backward as ops (#44408)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44408
See #44052 for context.
Test Plan: - `pytest test/test_autograd.py -v`
Reviewed By: mrshenli
Differential Revision: D23605504
Pulled By: zou3519
fbshipit-source-id: b9b1646d13caa6e536d08669c29bfc2ad8ff89a3