yolov5 How to get the class with highest confidence from a tensor?

Question

I want to obtain the class with the highest confidence. Heres the code that performs classification task:

names = [\'class A\', \'class B\', \'class C\']

def