gpt4 book ai didi

pytorch - PyTorch中torch.embedding的定义在哪里?

转载 作者:行者123 更新时间:2023-12-04 01:14:09 25 4
gpt4 key购买 nike

我试图了解 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 .

关于pytorch - PyTorch中torch.embedding的定义在哪里?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63913756/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com