Add __torch_function__ support for torch.nn.functional.embedding (#54478)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54478
Fixes #54292
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Test Plan: Imported from OSS
Reviewed By: jbschlosser
Differential Revision: D27264179
Pulled By: ezyang
fbshipit-source-id: cd267e2e668fdd8d7f958bf70a0b93e058ec7c23