Embedding implementation

I trying to understand how the torch.functional.F.embedding works with norm_type. I have looked at the source code: pytorch/functional.py at 46a32e075c0bdb54e3e2a08a370d785ee5c5a6cd · pytorch/pytorch · GitHub

Where does torch.embedding go on line 2024?

EDIT

Found it: pytorch/Embedding.cpp at 5b03ff0a09d43d721067e39da10aa23edc6997cd · pytorch/pytorch · GitHub

1 Like