pytorch-修改嵌入反向传播

Question

我想修改嵌入层的反向传播，但我不知道定义在哪里。

在嵌入功能的https://pytorch.org/docs/stable/_modules/torch/nn/functional.html中可用的定义中，它们调用C:\projects\python\test\mouse>python main.py 649 157 Traceback (most recent call last): File "main.py", line 10, in <module> main() File "main.py", line 6, in main mouse.move(0, 0) File "C:\projects\python\test\mouse\mouse.py", line 14, in move ms.move(dx - curX, dy - curY) TypeError: move() missing 1 required positional argument: 'dy' ，在此处应定义权重的更新方式。

所以我的问题是：
我在哪里可以找到torch.embedding的文档？

Answer 1

它在我的手电筒构建（版本4）this file中调用基础C函数。