def METHOD_NAME(output, target, topk=(1,)):
    """
    Compute top-k accuracy
    output: torch.Tensor
        shape (N, C) where N is the number of examples, C the number of classes.
        these are the logits.
    target: torch.Tensor
        shape (N,) where N is the number of examples. Groundtruth class id of each example.
    topk: tuple
        which topk to compute, e.g., topk=(1,5) will compute top-1 and top-5 accuracies
    Returns
    -------
    list of top-k accuracies in the same order as `topk`
    """
    # pred = output.topk(max(topk), 1, True, True)[1].t()
    pred = output.float().topk(max(topk), 1, True, True)[1].t()
    correct = pred.eq(target.view(1, -1).expand_as(pred))
    n = len(target)
    return [float(correct[:k].reshape(-1).float().sum(0, keepdim=True).cpu().numpy()) / n for k in topk]