如何解决PyTorch中的torch.embedding的定义在哪里?
我试图了解PyTorch如何创建嵌入并阅读torch.nn.functional.embedding
github link的源代码。
该函数返回torch.embedding(weight,input,padding_idx,scale_grad_by_freq,sparse)
的结果。
然后,我尝试理解torch.embedding
的定义,但是在GitHub存储库中找不到其源代码。在哪里?
解决方法
许多PyTorch的功能都是用C ++实现的。 torch.embedding
的入口点位于here。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。