Хуки позволяют наблюдать/модифицировать входы-выходы слоя без правки forward — для отладки, визуализации активаций, Grad-CAM.
Разбор
- Forward и full backward hooks.
- Не забывайте remove handle, иначе утечки.
- В TorchScript ограничены.
- Для прод-логики лучше явный код в forward.
Итог
Hooks — неинвазивная инструментализация слоёв.