def agg_metric()

in captum/robust/_core/metrics/attack_comparator.py [0:0]

• 6 lines of code
• 4 McCabe index (conditional complexity)

def agg_metric(inp):
    if isinstance(inp, Tensor):
        return inp.mean(dim=0)
    elif isinstance(inp, tuple):
        return tuple(agg_metric(elem) for elem in inp)
    return inp