pytorch - 修改嵌入反向传播

Question

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

在嵌入函数中https://pytorch.org/docs/stable/_modules/torch/nn/functional.html 的定义中，它们调用torch.embedding，这里应该定义如何更新权重。

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

Answer 1

它在我的 torch build(version 4) this file 中调用底层 C 函数。