В PyTorch, в чем разница между forward () и обычным методом?

Question

Чем реализация метода forward() пользовательского класса nn.Module отличается от добавления обычного метода к этому классу?

Я слышал, что метод forward() должен принимать и возвращать только тензоры, потому чтоPyTorch реализовал специальную обработку для ввода и вывода метода forward(). Но я попытался ввести / вывести нетензорные объекты с помощью метода forward() и реализовать модуль, у которого нет метода forward() (вместо этого есть несколько пользовательских методов, которые действуют как методы forward()). Оба способа работали хорошо.

Answer 1

forward() метод принимает параметры любого типа. Однако цель метода forward() заключается в инкапсуляции шагов прямого вычисления. forward() вызывается в функции __call__. В методе forward() PyTorch вызывает вложенную модель для выполнения прямого прохода.

Рекомендуется:

НЕ вызывать метод forward (x). Вы должны вызывать всю модель целиком, как в модели (x), для выполнения прогнозирования прямого прохода и вывода.

Что произойдет, если вы этого не сделаете?

Если вы позвонитеметод .forward() и наличие в вашей модели крючков , крючки не будут иметь никакого эффекта.