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

Question

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

5

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

- Hayato Maki

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- jodag · Accepted Answer

5

许多PyTorch函数是用C++实现的。torch.embedding的入口点位于这里。

- jodag

1

谢谢你的回答，jodag！你能告诉我如何确定代码的位置吗？ - Hayato Maki

2

@HayatoMaki 请查看此答案，其中描述了如何在pytorch中找到函数实现。 - jodag