这听起来可能有点基础,但是在运行下面的代码时,我想查看反向函数的源代码:
import torch.nn as nn
[...]
criterion = nn.CrossEntropyLoss()
loss = criterion(output, target)
loss.backward()
所以我去 PyTorch GitHub 找到了CrossEntropyLoss类,但没有定义任何后向函数。向上移动,CrossEntropyLoss 扩展了 _WeightedLoss >> _Loss >> Module 然后仍然没有。
那么,后向函数在哪里定义呢?