PyTorch 中定义的 Backward 函数在哪里?

数据挖掘 火炬
2021-10-03 02:57:45

这听起来可能有点基础,但是在运行下面的代码时,我想查看反向函数的源代码:

import torch.nn as nn

[...]
criterion = nn.CrossEntropyLoss()
loss = criterion(output, target)
loss.backward()

所以我去 PyTorch GitHub 找到了CrossEntropyLoss类,但没有定义任何后向函数。向上移动,CrossEntropyLoss 扩展了 _WeightedLoss >> _Loss >> Module 然后仍然没有。

那么,后向函数在哪里定义呢?