PyTorch 中 torch.embedding 的定义在哪里？

Question

我试图了解 PyTorch 如何创建嵌入并阅读torch.nn.functional.embedding github link的源代码。该函数返回 的结果torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)。然后，我尝试理解 的定义，torch.embedding但在 GitHub 存储库中找不到其源代码。它在哪里？

Answer 1

PyTorch 的许多功能都是用 C++ 实现的。的入口点torch.embedding位于此处。