def METHOD_NAME(exec_net, inputs, net):
    input_cpu = inputs.numpy()
    input_name = next(iter(net.inputs))
    output_name = next(iter(net.outputs))
    res = exec_net.infer(inputs={input_name: input_cpu})
    output = res[output_name]
    import torch
    torch_output = torch.from_numpy(output)
    return torch_output