torch.autograd.Function.forward

static Function.forward(ctx, *args, **kwargs) [source]

Выполняет операцию.

Этот метод должен быть переопределен всеми подклассами.

Он должен принимать контекст ctx в качестве первого аргумента, а затем любое количество аргументов (тензоры или другие типы).

Контекст может использоваться для хранения произвольных данных, которые затем могут быть получены во время обратного прохода. Тензоры не должны храниться напрямую в ctx (хотя это не обеспечивается в настоящее время для обратной совместимости). Вместо этого тензоры следует сохранять либо с помощью ctx.save_for_backward() если они предназначены для использования в backward (эквивалентно, vjp) или ctx.save_for_forward() если они предназначены для использования в jvp.

Тип возвращаемого значения:

Any