Method forward

intermediate_source/custom_function_conv_bn_tutorial.py:63–65

(ctx, X, weight)

class Conv2D(torch.autograd.Function):
    @staticmethod
    def forward(ctx, X, weight):
        ctx.save_for_backward(X, weight)
        return F.conv2d(X, weight)

    # Use @once_differentiable by default unless we intend to double backward
    @staticmethod