как получить ввод и окончательный вывод сети pytorch

Question

есть ли API, как получить узел ввода и вывода сети pytorch? Я пробовал model.features (), но это не поможет.

---

Пример: я получаю сеть pytorch и ее структуру в netron: network

Conv2d, MaxPool2d и Linear могут быть легко проанализированы. У меня проблемы с получением такой информации, как имя и размер входного и выходного узла.

Answer 1

Для получения ввода на конкретный слой означает вывод предыдущего слоя.

Таким образом, для получения любой информации во время прямого и обратного прохода, например, вывода определенного слоя c, градиента или, если вы хотите измените любой из них, который есть в pytorch, известный как Hooks .

Более подробную информацию можно найти здесь https://pytorch.org/tutorials/beginner/former_torchies/nnft_tutorial.html#forward -and-backward-function-hooks