What is a custom autograd Function in PyTorch?

Custom autograd Function allows defining forward and backward computation manually.

By subclassing torch.autograd.Function, you explicitly define forward pass and gradient computation, enabling custom operations not covered by built-in ops.