import torch


def aggregate_accuracy(test_logits_sample, test_labels):
    """
    Compute classification accuracy.
    """
    return torch.mean(torch.eq(test_labels, torch.argmax(test_logits_sample, dim=-1)).float())