def METHOD_NAME(self):
    labels = torch.tensor([0, 1, 0], dtype=torch.long)
    accuracy = TopKClassificationAccuracy(top_k=[1, 2])
    top1_acc, top2_acc = accuracy(logits=self.top_k_logits, labels=labels)
    assert accuracy.correct_counts_k.shape == torch.Size([2])
    assert accuracy.total_counts_k.shape == torch.Size([2])
    assert abs(top1_acc - 0.0) < 1e-3
    assert abs(top2_acc - 0.333) < 1e-3