我的PyTorch前进功能可以执行其他操作吗？ - python

通常，forward函数将一堆图层串在一起，并返回最后一层的输出。在返回之前，我可以在最后一层之后做一些其他处理吗？例如，一些标量乘法和通过.view重塑？ 我知道autograd会以某种方式找出梯度。因此，我不知道我的额外处理是否会导致这种情况。谢谢。 参考方案 pytorch通过张量的computational graph而不是通过函数跟踪渐变。只要您的 […]