def METHOD_NAME(ctx, input1):
    iH, iW = input1.size(2), input1.size(3)
    batch_size = input1.size(0)
    kH, kW = ctx.kernel_size
    patch_size = ctx.patch_size
    dH, dW = ctx.stride
    padH, padW = ctx.padding
    dilationH, dilationW = ctx.dilation
    dilatedKH = (kH - 1) * dilationH + 1
    dilatedKW = (kW - 1) * dilationW + 1
    oH = int((iH + 2 * padH - dilatedKH) / dH + 1)
    oW = int((iW + 2 * padW - dilatedKW) / dW + 1)
    output_size = (batch_size, patch_size, patch_size, oH, oW)
    return output_size