RuntimeError: ONNX export failed: Couldn't export Python operator qfn

I’m trying to convert a pytorch model to onnx, but the conversion gives the following error:

RuntimeError: ONNX export failed: Couldn’t export Python operator qfn

The definition for qfn is:

class qfn(torch.autograd.Function):
    def forward(ctx, input, k):
        n = float(2**k - 1)
        out = torch.round(input * n) / n
        return out

    def backward(ctx, grad_output):
        grad_input = grad_output.clone()
        return grad_input, None
1 Like